Skip to content

Latest commit

 

History

History
166 lines (103 loc) · 3.92 KB

README.md

File metadata and controls

166 lines (103 loc) · 3.92 KB

PPOCR-Rec

Current Support Platform

RK3566, RK3568, RK3588, RK3562, RK1808, RV1109, RV1126

Download ONNX model

Download link:

ppocrv4_rec.onnx

Download with shell command:

cd model
./download_model.sh

(Optional)PADDLE to ONNX

Please refer Paddle_2_ONNX.md

Model Convert

Usage:

cd python
python convert.py <onnx_model> <TARGET_PLATFORM> <dtype(optional)> <output_rknn_path(optional)>
# such as: python convert.py ../model/ppocrv4_rec.onnx rk3566
# output model will be saved as ../model/ppocrv4_rec.rknn

Description:

  • <onnx_model> should be the ONNX model path.
  • <TARGET_PLATFORM> could be specified as RK3562, RK3566, RK3568, RK3588, RK1808, RV1109, RV1126 according to board SOC version.
  • is optional, could be specified as fp, fp means no to do quantization, default is fp.
  • <output_rknn_path> is optional, used to specify the saving path of the RKNN model, default save path is ../model/ppocrv4_rec.rknn

Python Demo

Usage:

cd python

# Inference with ONNX model
python ppocr_rec.py --model_path <onnx_model>
# such as: python ppocr_rec.py --model_path ../model/ppocrv4_rec.onnx 

# Inference with RKNN model
python ppocrv4_rec.py --model_path <rknn_model> --target <TARGET_PLATFORM>
# such as: python ppocrv4_rec.py --model_path ../model/ppocrv4_rec.rknn --target rk3588

Description:

  • <TARGET_PLATFORM>: Specify NPU platform name. Such as 'rk3588'.

  • <onnx_model / rknn_model>: specified as the model path.

Android Demo

Note: RK1808, RV1109, RV1126 does not support Android.

Compiling && Building

Modify the path of Android NDK in 'build-android.sh'.

For example,

export ANDROID_NDK_PATH=~/opt/toolchain/android-ndk-r18b

Then, run this script:

./build-android.sh -t <TARGET_PLATFORM> -a arm64-v8a -d PPOCR-Rec

Please use the specific platform instead of <TARGET_PLATFORM> above.

Push all build output file to the board

Connect the USB port to PC, then push all demo files to the board.

adb root
adb push install/<TARGET_PLATFORM>_android_arm64-v8a/rknn_PPOCR-Rec_demo /data/

Running

adb shell
cd /data/rknn_PPOCR-Rec_demo

export LD_LIBRARY_PATH=./lib
./rknn_ppocr_rec_demo model/ppocrv4_rec.rknn model/test.png

Aarch64 Linux Demo

Compiling && Building

According to the target platform, modify the path of 'GCC_COMPILER' in 'build-linux.sh'.

export GCC_COMPILER=/opt/tools/prebuilts/gcc/linux-x86/aarch64/gcc-linaro-6.3.1-2017.05-x86_64_aarch64-linux-gnu/bin/aarch64-linux-gnu

Then, run the script:

./build-linux.sh  -t <TARGET_PLATFORM> -a aarch64 -d PPOCR-Rec

Please use the specific platform instead of <TARGET_PLATFORM> above.

Push all build output file to the board

Push install/<TARGET_PLATFORM>_linux_aarch64/rknn_PPOCR-Rec_demo to the board,

  • If use adb via the EVB board:
adb push install/<TARGET_PLATFORM>_linux_aarch64/rknn_PPOCR-Rec_demo /userdata/
  • For other boards, use the scp or other different approaches to push all files under install/<TARGET_PLATFORM>_linux_aarch64/rknn_PPOCR-Rec_demo to /userdata.

Please use the specific platform instead of <TARGET_PLATFORM> above.

Running

adb shell
cd /data/rknn_PPOCR-Rec_demo

export LD_LIBRARY_PATH=./lib
./rknn_ppocr_rec_demo model/ppocrv4_rec.rknn model/test.png

Note: Try searching the location of librga.so and add it to LD_LIBRARY_PATH if the librga.so is not found in the lib folder. Use the following command to add it to LD_LIBRARY_PATH.

export LD_LIBRARY_PATH=./lib:<LOCATION_LIBRGA>

Expected Results

This example will print the recognition result of the test image, as follows:

regconize result: JOINT, score=0.709082
  • Note: Different platforms, different versions of tools and drivers may have slightly different results.