RapidAI/paddleocr_convert:PaddleOCR中模型快速转换为ONNX格式

RapidAI/paddleocr_convert:PaddleOCR中模型快速转换为ONNX格式RapidAI/paddleocr_convert:PaddleOCR中模型快速转换为ONNX格式.

paddleocr_convert

RapidAI/paddleocr_convert:PaddleOCR中模型快速转换为ONNX格式 RapidAI/paddleocr_convert:PaddleOCR中模型快速转换为ONNX格式 PyPI RapidAI/paddleocr_convert:PaddleOCR中模型快速转换为ONNX格式 paddleocr_convert documentation

  • 本仓库主要是针对性地将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_PATH
                              The inference model url or local path of paddleocr.
                              e.g. https://paddleocr.bj.bcebos.com/PP-
                              OCRv3/chinese/ch_PP-OCRv3_det_infer.tar or
                              models/ch_PP-OCRv3_det_infer.tar
      -o SAVE_DIR, --save_dir SAVE_DIR
                              The directory of saving the model.
      -txt_path TXT_PATH, --txt_path TXT_PATH
                              The raw txt url or local txt path, if the model is
                              recognition 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 PaddleOCRModelConvert
    
    converter = 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 RapidOCR
      
      model_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中,便于后续分发。

今天的文章RapidAI/paddleocr_convert:PaddleOCR中模型快速转换为ONNX格式分享到此就结束了,感谢您的阅读。

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
如需转载请保留出处:http://bianchenghao.cn/22186.html

(0)
编程小号编程小号

相关推荐

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注