# PaddleOCRModelConverter

**Repository Path**: encv/PaddleOCRModelConverter

## Basic Information

- **Project Name**: PaddleOCRModelConverter
- **Description**: Forked from: https://github.com/RapidAI/PaddleOCRModelConverter
- **Primary Language**: Unknown
- **License**: Apache-2.0
- **Default Branch**: encv
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2022-06-09
- **Last Updated**: 2022-06-23

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# PaddleOCR v3 相关模型转换

本仓库主要是针对性地将[PaddleOCR/release/v2.5](https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.5/doc/doc_ch/models_list.md)转换为onnx模型

* [Open in AI Stuido](https://aistudio.baidu.com/aistudio/projectdetail/3974957?_=1652277622041&shared=1)

* [github: PaddleOCRModelConverter](https://github.com/RapidAI/PaddleOCRModelConverter)

* [gitee: encv/PaddleOCRModelConverter](https://gitee.com/encv/PaddleOCRModelConverter)

## 1. 运行环境
- 操作系统： Linux / Mac
- 安装依赖包:
    ```bash
    pip install -r requirements.txt -i https://pypi.douban.com/simple/
    ```
- 测试可以成功转换的库版本如下：
  ```txt
  onnx                        1.8.0
  onnxruntime                 1.10.0
  opencv-python               4.2.0.32
  paddle2onnx                 0.9.0
  paddlepaddle-gpu            2.0.2.post100
  ```

## 2. PaddleOCR推理模型-下载地址

目前版本并未测试稳定。所以建议使用前，更新PaddleOCR的官方推理模型地址:

[gitee: PP-OCR系列模型列表（V3，2022年4月28日更新）](https://gitee.com/paddlepaddle/PaddleOCR/blob/release/2.5/doc/doc_ch/models_list.md)

手动将模型解压在`pretrained_models`目录下，脚本将不再覆盖下载。

## 3. 使用步骤
1. 直接运行一下对应脚本即可
    - 推理模型转换
        ```shell
        $ bash det_inference_to_onnx.sh
        # bash cls_inference_to_onnx.sh  # 这个模型转换后误差较大
        $ bash rec_inference_to_onnx.sh
        ```
    - 预训练模型转换
        ```shell
        $ bash det_pretrain_to_onnx.sh
        $ bash cls_pretrain_to_onnx.sh
        $ bash rec_pretrain_to_onnx.sh
        ```
    - 如果遇到`rec_inference_to_onnx.sh: line 3: $'\r': command not found`类似错误
        ```bash
        $ vi rec_inference_to_onnx.sh

        # vi中执行以下命令即可
        # :set ff=unix
        # :wq
        ```
2. 最终转换后的模型会在`convert_model`目录下
3. 可以移步到[RapidOCR部分的python目录](https://github.com/RapidAI/RapidOCR/tree/main/python/onnxruntime_infer)，替换相应模型即可