65 lines
1.8 KiB
Markdown
65 lines
1.8 KiB
Markdown
|
|
# paddleocr模型转换成onnx模型后,利用ONNX模型进行推理
|
|||
|
|
## 1、安装paddle2onnx
|
|||
|
|
```angular2html
|
|||
|
|
pip install paddle2onnx
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
## 2、下载paddleocr模型文件
|
|||
|
|
```angular2html
|
|||
|
|
!wget https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_cls_infer.tar
|
|||
|
|
!wget https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_server_v2.0_rec_infer.tar
|
|||
|
|
!wget https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_server_v2.0_det_infer.tar
|
|||
|
|
```
|
|||
|
|
## 3、解压模型文件
|
|||
|
|
```angular2html
|
|||
|
|
!tar -xvf /home/aistudio/onnx_pred/models/ch_ppocr_mobile_v2.0_cls_infer.tar
|
|||
|
|
!tar -xvf /home/aistudio/onnx_pred/models/ch_ppocr_server_v2.0_det_infer.tar
|
|||
|
|
!tar -xvf /home/aistudio/onnx_pred/models/ch_ppocr_server_v2.0_rec_infer.tar
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
## 4、将paddleocr模型转成onxx模型
|
|||
|
|
```angular2html
|
|||
|
|
paddle2onnx --model_dir ./ch_ppocr_server_v2.0_rec_infer \
|
|||
|
|
--model_filename inference.pdmodel \
|
|||
|
|
--params_filename inference.pdiparams \
|
|||
|
|
--save_file ./ch_ppocr_server_v2.0_rec.onnx \
|
|||
|
|
--opset_version 11 \
|
|||
|
|
--enable_onnx_checker True
|
|||
|
|
|
|||
|
|
|
|||
|
|
paddle2onnx --model_dir ./ch_ppocr_server_v2.0_det_infer \
|
|||
|
|
--model_filename inference.pdmodel \
|
|||
|
|
--params_filename inference.pdiparams \
|
|||
|
|
--save_file ./ch_ppocr_server_v2.0_det.onnx \
|
|||
|
|
--opset_version 11 \
|
|||
|
|
--enable_onnx_checker True
|
|||
|
|
|
|||
|
|
|
|||
|
|
paddle2onnx --model_dir ./ch_ppocr_mobile_v2.0_cls_infer \
|
|||
|
|
--model_filename inference.pdmodel \
|
|||
|
|
--params_filename inference.pdiparams \
|
|||
|
|
--save_file ./ch_ppocr_mobile_v2.0_cls.onnx \
|
|||
|
|
--opset_version 11 \
|
|||
|
|
--enable_onnx_checker True
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
## 5、安装onnx
|
|||
|
|
```angular2html
|
|||
|
|
pip install onnx==1.14.0
|
|||
|
|
pip install onnxruntime-gpu==1.14.1
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
## 6、模型推理
|
|||
|
|
```angular2html
|
|||
|
|
import cv2
|
|||
|
|
model = ONNXPaddleOcr()
|
|||
|
|
|
|||
|
|
img = cv2.imread('./1.jpg')
|
|||
|
|
|
|||
|
|
# ocr识别结果
|
|||
|
|
result = model.ocr(img)
|
|||
|
|
print(result)
|
|||
|
|
|
|||
|
|
# 画box框
|
|||
|
|
sav2Img(img, result)
|
|||
|
|
```
|