RapidAI/paddleocr_convert

• 本仓库主要是针对性地将PaddleOCR中推理模型转换为ONNX格式。
• 注意：
  • 输入：推理模型的url或者本地tar路径
  • 输出：转换后的ONNX模型
  • 如果是识别模型，需要提供对应字典的原始txt路径（打开github中txt文件，点击右上角raw之后的路径），用来将字典写入到onnx模型中
  • ！！！需要搭配RapidOCR中相关推理代码使用

使用步骤

1. 安装paddleocr_convert

   pip install paddleocr_convert
   

2. 命令行使用

   • 用法:

     $ paddleocr_convert -h
     usage: paddleocr_convert [-h] [-p MODEL_PATH] [-o SAVE_DIR][-txt_path TXT_PATH]optional arguments:
     -h, --help            show this help message and exit
     -p MODEL_PATH, --model_path MODEL_PATHThe inference model url or local path of paddleocr.e.g. https://paddleocr.bj.bcebos.com/PP-OCRv3/chinese/ch_PP-OCRv3_det_infer.tar ormodels/ch_PP-OCRv3_det_infer.tar
     -o SAVE_DIR, --save_dir SAVE_DIRThe directory of saving the model.
     -txt_path TXT_PATH, --txt_path TXT_PATHThe raw txt url or local txt path, if the model isrecognition model.
     

   • 示例:

     # online
     $ paddleocr_convert -p https://paddleocr.bj.bcebos.com/PP-OCRv3/chinese/ch_PP-OCRv3_det_infer.tar \-o models$ paddleocr_convert -p https://paddleocr.bj.bcebos.com/PP-OCRv3/chinese/ch_PP-OCRv3_rec_infer.tar \-o models \-txt_path https://raw.githubusercontent.com/PaddlePaddle/PaddleOCR/release/2.6/ppocr/utils/ppocr_keys_v1.txt# offline
     $ paddleocr_convert -p models/ch_PP-OCRv3_det_infer.tar \-o models$ paddleocr_convert -p models/ch_PP-OCRv3_rec_infer.tar \-o models \-txt_path models/ppocr_keys_v1.txt
     

3. 脚本使用

   from paddleocr_convert import PaddleOCRModelConvertconverter = PaddleOCRModelConvert()save_dir = 'models'# online
   url = 'https://paddleocr.bj.bcebos.com/PP-OCRv3/chinese/ch_PP-OCRv3_rec_infer.tar'
   txt_url = 'https://raw.githubusercontent.com/PaddlePaddle/PaddleOCR/release/2.6/ppocr/utils/ppocr_keys_v1.txt'converter(url, save_dir, txt_path=txt_url)# offline
   model_path = 'models/ch_PP-OCRv3_rec_infer.tar'
   txt_path = 'models/ppocr_keys_v1.txt'
   converter(model_path, save_dir, txt_path=txt_path)
   

4. 使用模型方法：

   • 假设要用日文识别模型，且已经转好，路径为local/models/japan.onnx
   1. 安装rapidocr_onnxruntime库

      pip install rapidocr_onnxruntime
      

   2. 脚本使用

      from rapidocr_onnxruntime import RapidOCRmodel_path = 'local/models/japan.onnx'
      engine = RapidOCR(rec_model_path=model_path)img = '1.jpg'
      result, elapse = engine(img)
      

   3. 命令行使用

      $ rapidocr_onnxruntime -img 1.jpg --rec_model_path local/models/japan.onnx
      

更新日志

• 2023-03-05 v0.0.4~7 update:

  • 支持对本地的模型和字典转写
  • 优化内部逻辑和错误反馈

• 2023-02-28 v0.0.3 update:

  • 添加对不是动态输入的模型自动更改为动态输入的设置

• 2023-02-27 v0.0.2 update:

  • 将转换模型代码封装成包，便于自助转模型

• 2022-08-15 v0.0.1 update:

  • 将识别模型的字典写入到onnx模型中的meta中，便于后续分发。