PaddleOCRv5模型导出ONNX格式的解决方案

问题背景

在使用PaddleOCRv5进行模型导出为ONNX格式时，部分用户遇到了转换失败的问题。特别是在Windows系统环境下，按照某些教程操作时会出现版本不兼容的错误提示。

错误分析

主要错误信息显示："The paddlepaddle version should not be less than 3.0.0.dev20250426"，这表明当前安装的PaddlePaddle版本不符合要求。错误发生在使用paddle2onnx工具进行模型转换时。

解决方案

Windows系统下的解决方案

1. 安装正确的PaddlePaddle版本： 需要安装开发版的PaddlePaddle，可以使用以下命令安装：

   python -m pip install --pre paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu118/
   

2. 修改源代码： 在导出模型前，需要修改PP-OCRv5的源代码文件：

   • 找到ppocr/utils/export_model.py文件
   • 将第389行的条件判断修改为if False:
   • 修改完成后保存文件

3. 执行模型导出： 使用以下命令导出模型：

   python tools/export_model.py -c ./configs/PP-OCRv5_server_det.yml -o Global.pretrained_model=./models/PP-OCRv5_server_det_pretrained.pdparams Global.save_inference_dir=./onnxmodels/PP-OCRv5_server_det
   

4. 转换为ONNX格式： 使用paddle2onnx工具进行转换：

   paddle2onnx --model_dir ./onnxmodels/PP-OCRv5_server_det --model_filename inference.pdmodel --params_filename inference.pdiparams --save_file model.onnx --opset_version 13 --enable_onnx_checker True
   

环境配置建议

• 操作系统：Windows 11
• PaddlePaddle版本：3.0.0（GPU版）
• PaddleOCR版本：3.0.1
• paddle2onnx版本：1.3.1

注意事项

1. 完成ONNX导出后，建议将修改过的源代码恢复原状，以避免影响其他功能。
2. 不同版本的PaddleOCR可能需要不同的处理方式，建议使用官方推荐的版本组合。
3. 如果使用GPU版本，请确保CUDA环境配置正确。

总结

通过正确配置环境和修改源代码，可以成功将PaddleOCRv5模型导出为ONNX格式。这一过程需要注意版本兼容性和系统环境差异，特别是在Windows系统上操作时。建议用户在操作前仔细检查环境配置，并按照步骤顺序执行。