PaddleOCR用onnx推理代码并精度对齐

训练脚本:

python ./tools/train.py -c ./configs/rec/SVTRv2/rec_repsvtr_gtc.yml
 

SVTRV2版本导出:

python tools/export_model.py -c ./configs/rec/SVTRv2/rec_repsvtr_gtc.yml -o Global.pretrained_model=output/rec_repsvtr_gtc/best_accuracy Global.save_inference_dir=./inference/rec_repsvtr_infer

导出onnx:

paddle2onnx --model_dir ./inference/rec_repsvtr_infer/ --model_file inference.pdmodel --params_filename inference.pdiparams --save_file ./inference/rec_repsvtr_infer/model.onnx --opset_version 11 --enable_onnx_checker True

onnxruntime推理:

python3.7 tools/infer/predict_system.py --use_gpu=False --use_onnx=True \

--det_model_dir=./inference/det_onnx/model.onnx  \

--rec_model_dir=./inference/rec_onnx/model.onnx  \

--cls_model_dir=./inference/cls_onnx/model.onnx  \

--image_dir=doc/imgs_en/img_12.jpg \</

相关推荐

  1. onnx推理python代码

    2024-07-20 03:16:02       53 阅读
  2. yolov8目标检测-onnx模型推理

    2024-07-20 03:16:02       39 阅读
  3. yolov8pt转onnx(官方代码

    2024-07-20 03:16:02       14 阅读
  4. TensorRT加速推理入门-1:Pytorch转ONNX

    2024-07-20 03:16:02       42 阅读
  5. pytorch导出为onnx,使用onnxruntime进行推理

    2024-07-20 03:16:02       35 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-20 03:16:02       52 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-20 03:16:02       54 阅读
  3. 在Django里面运行非项目文件

    2024-07-20 03:16:02       45 阅读
  4. Python语言-面向对象

    2024-07-20 03:16:02       55 阅读

热门阅读

  1. 数据的守护者:深入解析 Elasticsearch 的副本机制

    2024-07-20 03:16:02       18 阅读
  2. Linux C++ 058-设计模式之解释器模式

    2024-07-20 03:16:02       21 阅读
  3. Swagger生成Api文档的增强解决方案--knife4j

    2024-07-20 03:16:02       22 阅读
  4. 虫虫老师---义务教育核心课程改革

    2024-07-20 03:16:02       14 阅读
  5. 面试题 16.07. 最大数值

    2024-07-20 03:16:02       16 阅读
  6. 【基础算法】排序

    2024-07-20 03:16:02       14 阅读
  7. 说说Vue2.0和Vue3.0有什么区别

    2024-07-20 03:16:02       17 阅读
  8. kubernetes学习日志(六)

    2024-07-20 03:16:02       12 阅读
  9. JWT身份验证、授权介绍、应用场景和示例代码

    2024-07-20 03:16:02       18 阅读
  10. VUE3【实用教程】(2024最新版)

    2024-07-20 03:16:02       19 阅读
  11. LLM推理需要占用多少显存

    2024-07-20 03:16:02       16 阅读