架构师_程序员_码农网

 找回密码
 注册[Register]

QQ登录

只需一步,快速开始

搜索
查看: 169|回复: 1

将 paddle 模型转换成 onnx 模型格式

[复制链接]
发表于 2025-1-22 08:51:14 | 显示全部楼层 |阅读模式
需求:将 paddle 模型转换成 onnx 模型格式,使用 onnxruntime 进行推理,以 paddleocr 为例。

百度官方提供了 Paddle2ONNX 转换工具,地址:https://github.com/PaddlePaddle/Paddle2ONNX

官方工具不太好用,而且缺少一些功能,本文使用 PaddleOCRModelConvert 工具进行转换。例如:将文本识别模型对应的字典文件写入到 onnx 模型中,如下图:

QQ截图20250122084306.jpg

模型下载:https://github.com/PaddlePaddle/PaddleOCR/blob/40c56628fda416e1c8710eb19e4b260536902520/doc/doc_ch/models_list.md

首先,使用 conda 创建环境,命令如下:


转换命令如下:


如下图:

QQ截图20250122085047.jpg

PaddleOCRModelConvert 地址:https://github.com/RapidAI/PaddleOCRModelConvert

参考:
https://github.com/RapidAI/PaddleOCRModelConvert/blob/main/paddleocr_convert/main.py
https://github.com/PaddlePaddle/PaddleOCR/tree/main/deploy/paddle2onnx




上一篇:【转】深度学习 DBNet 的基本原理
下一篇:两款前端图像(图片)比较器
码农网,只发表在实践过程中,遇到的技术难题,不误导他人。
 楼主| 发表于 2025-1-23 10:48:48 | 显示全部楼层
码农网,只发表在实践过程中,遇到的技术难题,不误导他人。
您需要登录后才可以回帖 登录 | 注册[Register]

本版积分规则

免责声明:
码农网所发布的一切软件、编程资料或者文章仅限用于学习和研究目的;不得将上述内容用于商业或者非法用途,否则,一切后果请用户自负。本站信息来自网络,版权争议与本站无关。您必须在下载后的24个小时之内,从您的电脑中彻底删除上述内容。如果您喜欢该程序,请支持正版软件,购买注册,得到更好的正版服务。如有侵权请邮件与我们联系处理。

Mail To:help@itsvse.com

QQ|手机版|小黑屋|架构师 ( 鲁ICP备14021824号-2 )|网站地图

GMT+8, 2025-2-18 17:29

Powered by Discuz! X3.4

Copyright © 2001-2021, Tencent Cloud.

快速回复 返回顶部 返回列表