推理第三方模型
第三方模型支持列表¶
MindOCR可以支持第三方模型（如PaddleOCR等）的推理，本文档展示了已适配的模型列表。 性能测试基于Ascend310P，部分模型暂无测试数据集。
本页面所有结果均基于MindSpore2.2.14版本310P3芯片。
文本检测¶
| 名称 | 模型 | 骨干网络 | 数据集 | F-score(%) | FPS | 来源 | 配置文件 | 下载 | 参考链接 | ONNX | Data Shape (NCHW) | Lite convert config txt |
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| ch_pp_det_OCRv4 | DBNet | MobileNetV3 | / | / | / | PaddleOCR | yaml | infer model | ch_PP-OCRv4_det | onnx | (-1,3,-1,-1) | config txt |
| ch_pp_server_det_v2.0 | DBNet | ResNet18_vd | MLT17 | 48.28 | 12.68 | PaddleOCR | yaml | infer model | ch_ppocr_server_v2.0_det | onnx | (1,3,736,1280) | config txt |
| ch_pp_det_OCRv3 | DBNet | MobileNetV3 | MLT17 | 35.02 | 23.91 | PaddleOCR | yaml | infer model | ch_PP-OCRv3_det | onnx | (-1,3,-1,-1) | config txt |
| ch_pp_det_OCRv2 | DBNet | MobileNetV3 | MLT17 | 44.80 | 13.00 | PaddleOCR | yaml | infer model | ch_PP-OCRv2_det | onnx | (1,3,736,1280) | config txt |
| ch_pp_mobile_det_v2.0_slim | DBNet | MobileNetV3 | MLT17 | 33.48 | 12.36 | PaddleOCR | yaml | infer model | ch_ppocr_mobile_slim_v2.0_det | onnx | (1,3,736,1280) | config txt |
| ch_pp_mobile_det_v2.0 | DBNet | MobileNetV3 | MLT17 | 33.36 | 12.77 | PaddleOCR | yaml | infer model | ch_ppocr_mobile_v2.0_det | onnx | (1,3,736,1280) | config txt |
| en_pp_det_OCRv3 | DBNet | MobileNetV3 | IC15 | 43.73 | 40.53 | PaddleOCR | yaml | infer model | en_PP-OCRv3_det | onnx | (-1,3,-1,-1) | config txt |
| ml_pp_det_OCRv3 | DBNet | MobileNetV3 | MLT17 | 68.79 | 20.12 | PaddleOCR | yaml | infer model | ml_PP-OCRv3_det | onnx | (-1,3,-1,-1) | config txt |
| en_pp_det_dbnet_resnet50vd | DBNet | ResNet50_vd | IC15 | 79.25 | 23.71 | PaddleOCR | yaml | infer model | DBNet | onnx | (1,3,736,1280) | config txt |
| en_pp_det_psenet_resnet50vd | PSE | ResNet50_vd | IC15 | 82.01 | 10.22 | PaddleOCR | yaml | train model | PSE | onnx | (1,3,736,1280) | config txt |
| en_pp_det_east_resnet50vd | EAST | ResNet50_vd | IC15 | 84.80 | 23.32 | PaddleOCR | yaml | train model | EAST | onnx | (1,3,736,1280) | config txt |
| en_pp_det_sast_resnet50vd | SAST | ResNet50_vd | IC15 | 86.28 | 17.45 | PaddleOCR | yaml | train model | SAST | onnx | (1,3,-1,-1) | config txt |
注意：在使用en_pp_det_psenet_resnet50vd模型进行推理时，需要使用以下命令修改onnx文件
python deploy/models_utils/onnx_optim/insert_pse_postprocess.py \
--model_path=./pse_r50vd.onnx \
--binary_thresh=0.0 \
--scale=1.0
文本识别¶
| 名称 | 模型 | 骨干网络 | 数据集 | Acc(%) | FPS | 来源 | 字典文件 | 配置文件 | 下载 | 参考链接 | ONNX | Data Shape (NCHW) | Lite convert config txt |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| ch_pp_rec_OCRv4 | CRNN | MobileNetV1Enhance | / | / | 310.71 | PaddleOCR | ppocr_keys_v1.txt | yaml | infer model | ch_PP-OCRv4_rec | onnx | (1,3,32,100) | config txt |
| ch_pp_server_rec_v2.0 | CRNN | ResNet34 | / | / | 259.82 | PaddleOCR | ppocr_keys_v1.txt | yaml | infer model | ch_ppocr_server_v2.0_rec | onnx | (1,3,32,100) | config txt |
| ch_pp_rec_OCRv3 | SVTR | MobileNetV1Enhance | / | / | 430.02 | PaddleOCR | ppocr_keys_v1.txt | yaml | infer model | ch_PP-OCRv3_rec | onnx | (1,3,32,100) | config txt |
| ch_pp_rec_OCRv2 | CRNN | MobileNetV1Enhance | / | / | 321.91 | PaddleOCR | ppocr_keys_v1.txt | yaml | infer model | ch_PP-OCRv2_rec | onnx | (1,3,32,100) | config txt |
| ch_pp_mobile_rec_v2.0 | CRNN | MobileNetV3 | / | / | 278.21 | PaddleOCR | ppocr_keys_v1.txt | yaml | infer model | ch_ppocr_mobile_v2.0_rec | onnx | (1,3,32,100) | config txt |
| en_pp_rec_OCRv3 | SVTR | MobileNetV1Enhance | IC15 | 49.83 | 580.88 | PaddleOCR | en_dict.txt | yaml | infer model | en_PP-OCRv3_rec | onnx | (1,3,32,100) | config txt |
| en_pp_mobile_rec_number_v2.0_slim | CRNN | MobileNetV3 | / | / | 229.21 | PaddleOCR | en_dict.txt | yaml | infer model | en_number_mobile_slim_v2.0_rec | onnx | (1,3,32,320) | config txt |
| en_pp_mobile_rec_number_v2.0 | CRNN | MobileNetV3 | IC15 | 46.08 | 228.97 | PaddleOCR | en_dict.txt | yaml | infer model | en_number_mobile_v2.0_rec | onnx | (1,3,32,320) | config txt |
| korean_pp_rec_OCRv3 | SVTR | MobileNetV1Enhance | / | / | 511.92 | PaddleOCR | korean_dict.txt | yaml | infer model | korean_PP-OCRv3_rec | onnx | (1,3,32,100) | config txt |
| japan_pp_rec_OCRv3 | SVTR | MobileNetV1Enhance | / | / | 477.32 | PaddleOCR | japan_dict.txt | yaml | infer model | japan_PP-OCRv3_rec | onnx | (1,3,32,100) | config txt |
| chinese_cht_pp_rec_OCRv3 | SVTR | MobileNetV1Enhance | / | / | 390.15 | PaddleOCR | chinese_cht_dict.txt | yaml | infer model | chinese_cht_PP-OCRv3_rec | onnx | (1,3,32,100) | config txt |
| te_pp_rec_OCRv3 | SVTR | MobileNetV1Enhance | / | / | 582.82 | PaddleOCR | te_dict.txt | yaml | infer model | te_PP-OCRv3_rec | onnx | (1,3,32,100) | config txt |
| ka_pp_rec_OCRv3 | SVTR | MobileNetV1Enhance | / | / | 589.41 | PaddleOCR | ka_dict.txt | yaml | infer model | ka_PP-OCRv3_rec | onnx | (1,3,32,100) | config txt |
| ta_pp_rec_OCRv3 | SVTR | MobileNetV1Enhance | / | / | 587.30 | PaddleOCR | ta_dict.txt | yaml | infer model | ta_PP-OCRv3_rec | onnx | (1,3,32,100) | config txt |
| latin_pp_rec_OCRv3 | SVTR | MobileNetV1Enhance | / | / | 584.57 | PaddleOCR | latin_dict.txt | yaml | infer model | latin_PP-OCRv3_rec | onnx | (1,3,32,100) | config txt |
| arabic_pp_rec_OCRv3 | SVTR | MobileNetV1Enhance | / | / | 578.28 | PaddleOCR | arabic_dict.txt | yaml | infer model | arabic_PP-OCRv3_rec | onnx | (1,3,32,100) | config txt |
| cyrillic_pp_rec_OCRv3 | SVTR | MobileNetV1Enhance | / | / | 573.57 | PaddleOCR | cyrillic_dict.txt | yaml | infer model | cyrillic_PP-OCRv3_rec | onnx | (1,3,32,100) | config txt |
| devanagari_pp_rec_OCRv3 | SVTR | MobileNetV1Enhance | / | / | 589.43 | PaddleOCR | devanagari_dict.txt | yaml | infer model | devanagari_PP-OCRv3_rec | onnx | (1,3,32,100) | config txt |
| en_pp_rec_crnn_resnet34vd | CRNN | ResNet34_vd | IC15 | 66.35 | 392.97 | PaddleOCR | ic15_dict.txt | yaml | infer model | CRNN | onnx | (1,3,32,160) | config txt |
| en_pp_rec_rosetta_resnet34vd | Rosetta | Resnet34_vd | IC15 | 64.52 | 487.68 | PaddleOCR | ic15_dict.txt | yaml | infer model | Rosetta | onnx | (1,3,32,160) | config txt |
| en_pp_rec_vitstr_vitstr | ViTSTR | ViTSTR | IC15 | 68.42 | 336.53 | PaddleOCR | EN_symbol_dict.txt | yaml | train model | ViTSTR | onnx | (1,1,224,224) | config txt |
文本方向分类¶
| 名称 | 模型 | 数据集 | Acc(%) | FPS | 来源 | 配置文件 | 下载 | 参考链接 | ONNX | Data Shape (NCHW) | Lite convert config txt |
|---|---|---|---|---|---|---|---|---|---|---|---|
| ch_pp_mobile_cls_v2.0 | MobileNetV3 | / | / | / | PaddleOCR | yaml | infer model | ch_ppocr_mobile_v2.0_cls | onnx | (1,3,48,192) | config txt |