作者:殷忆枫

RV1126笔记三十八:PaddleOCR部署到RV1126

若该文为原创文章,转载请注明原文出处。

一、环境

1、硬件:正点原子RV1126开发板

2、环境:ubuntu16.04 

二、模型转换

训练后的模型不能直接使用在RV1126,需要转换一下模型

1、PaddlePaddle的模型转成推理模型

在前面有提过了;

训练模型保存为用于部署的推理模型

保存路径:output文件夹内

python tools/export_model.py -c "./configs/det/ch_ppocr_v2.0/ch_det_mv3_db_v2.0.yml_car_plate.yml" -o Global.pretrained_model="./output/ch_db_mv3/latest" Global.save_inference_dir="./output/"

2、转成onnx模型

PaddleOCR有一个专门的工具 paddle2onnx,

更多详情请参考Paddle2ONNX的Github主页

安装方法有两种:

# PIP 安装
$ pip install paddle2onnx
# 源码安装
$ git clone https://github.com/paddlepaddle/paddle2onnx
$ cd paddle2onnx 
$ python setup.py install

安装后把推理模型转成onnx

paddle2onnx --model
lock