关于Pytorch导出ONNX格式模型的Tips

科技 2024-11-30 22:05 江苏

点击上方蓝字关注我们

微信公众号：OpenCV学堂
关注获取更多计算机视觉与深度学习知识

错误信息

昨天有个课程学员跟我反馈说YOLOv5导出的ONNX格式模型，OpenCV DNN无法读取的问题，我以为是它使用的OpenCV版本不是4.7以后的，当他告诉我使用的OpenCV DNN版本是4.10.x时候，我实在无法想出是什么问题，于是今天请他把模型文件跟代码都发我了，我直接运行了一下，发现的确会有读取错误，错误信息如下：

发现的确是无法加载，但是我用本地的我之前导出来的YOLOv5的模型，发现加载没有问题，没有错误，于是我意识到肯定导出的ONNX格式的模型文件有相关的差异，用Netron打开以后对比信息如下：

对比发现只有这两个地方不一样，查找官方资料说法如下：

意思是导出的模型不是JIT模式，但是我不知道main graph是怎么导出来的，是不是Pytorch2.4已经支持这种模式导出了，

导出ONNX

从Pytorch导出ONNX格式模型都是基于JIT模式（推理模式）如果模型不是JIT模式，Pytorch的ONNX导出函数会先内部把模型转换为JIT模式，然后再导出ONNX格式模型。这个就是Pytorch导出ONNX格式的Tracing and Scripting过程，官方的解释原文如下：

代码示例：

OpenCV4系统化学习

深度学习系统化学习

推荐阅读

OpenCV4.8+YOLOv8对象检测C++推理演示

ZXING+OpenCV打造开源条码检测应用

攻略 | 学习深度学习只需要三个月的好方法

三行代码实现 TensorRT8.6 C++ 深度学习模型部署

实战 | YOLOv8+OpenCV 实现DM码定位检测与解析

对象检测边界框损失 – 从IOU到ProbIOU

初学者必看 | 学习深度学习的五个误区

OpenCV学堂

三本书《Java数字图像处理-编程技巧与应用实践》、《OpenCV Android开发实战》、《OpenCV4应用开发-入门、进阶与工程化实践》作者。OpenCV实验大师平台软件作者，OpenCV开发专家、OpenCV研习社创始人。

最新文章

QT5+ 工业相机 + OpenCV工作流集成演示

Pytorch轻松实现经典视觉任务

正式发布 | QT5 + OpenCV4.8从入门到实战路线图

十种主流模型C++部署演示

OpenCV案例 | 生物检测-提取脂肪球

注册NVIDIA DLI新用户，免费领取价值600元的课程，写评语有好礼

视频一键拆分PS层！DeepMind新模型效果碾压同级，物体、背景完美分离，还能脑补

OpenCV4+QT5 深度学习应用开发演示

LSTM之父：我也是注意力之父！1991年就发表线性复杂度，遥遥领先Transformer 26年

OpenCV4 | 怎么学才能找到工作

超图计算+目标检测，性能新SOTA！清华发布Hyper-YOLO：用超图捕捉高阶视觉关联

Pytorch轻松实现经典视觉任务

OpenCV4+QT5 深度学习应用开发演示

YOLOv8与YOLO11自定义数据集迁移学习效果对比

OpenCV实现各种图形绘制、图像调色

推荐 | OpenCV算法开发能力暴涨十倍的方法

YOLOv8 + OpenVINO + QT5 打造电子围栏预警系统

推荐 | OpenCV算法开发能力暴涨十倍的方法

编程高薪神话末日来临？23届计算机本科就业率狂掉3.2%，AI技能成救命稻草

福利 | OpenCV4+深度学习+QT5全系课程

YOLOv8+PyQT5打造安全帽检测预警应用

Pytorch轻松实现经典视觉任务

关于Pytorch导出ONNX格式模型的Tips

正式发布 | QT5 + OpenCV4.8从入门到实战路线图

YOLOv8与YOLO11自定义数据集迁移学习效果对比

正式发布 | QT5 + OpenCV4.8从入门到实战路线图

C#与OpenCV C++导出DLL调用与数据交互

十种主流模型C++部署演示

Claude 3.5两小时暴虐50多名专家，编程10倍速飙升！但8小时曝出惊人短板

Pytorch轻松实现经典视觉任务

OpenCV案例 | 角点检测

Pytorch轻松实现经典视觉任务

正式发布 | QT5 + OpenCV4.8从入门到实战路线图

YOLOv8的 Web端网页推理演示

常识 | 机器视觉算法如何选择相机分辨率

Pytorch轻松实现经典视觉任务

Torchvision框架学习之FCOS模型及其训练

福利 | OpenCV4+深度学习+QT5全系课程

图像编辑器 Monica 之实现好玩的人脸替换功能

OpenCV4 | 怎么学才能找到工作

Numpy包从1.x升级到2.x以后的几个注意点

福利 | OpenCV4+深度学习+QT5全系课程

OpenCV案例 | 基于轮廓分析对象提取

AlphaFold3重磅开源，诺奖级AI颠覆世界！GitHub斩获1.8k星，本地即可部署

Pytorch轻松实现经典视觉任务

OpenCV4 | 怎么学才能找到工作

十种主流模型C++部署演示

深度学习入门必学之Pytorch框架实战课程

正式发布 | QT5 + OpenCV4.8从入门到实战路线图

福利 | OpenCV4+深度学习+QT5全系课程

分类

时事

民生

政务

教育

文化

科技

财富

体娱

健康

情感

旅行

百科

职场

楼市

企业

乐活

学术

汽车

时尚

创业

美食

幽默

美体

文摘

原创标签

时事社会财经军事教育体育科技汽车科学房产搞笑综艺明星音乐动漫游戏时尚健康旅游美食生活摄影宠物职场育儿情感小说曲艺文化历史三农文学娱乐电影视频图片新闻宗教电视剧纪录片广告创意壁纸头像心灵鸡汤星座命理教育培训艺术文化金融财经健康医疗美妆时尚餐饮美食母婴育儿社会新闻工业农业时事政治星座占卜幽默笑话独立短篇连载作品文化历史科技互联网

发布位置

广东北京山东江苏河南浙江山西福建河北上海四川陕西湖南安徽湖北内蒙古江西云南广西甘肃辽宁黑龙江贵州新疆重庆吉林天津海南青海宁夏西藏香港澳门台湾美国加拿大澳大利亚日本新加坡英国西班牙新西兰韩国泰国法国德国意大利缅甸菲律宾马来西亚越南荷兰柬埔寨俄罗斯巴西智利卢森堡芬兰瑞典比利时瑞士土耳其斐济挪威朝鲜尼日利亚阿根廷匈牙利爱尔兰印度老挝葡萄牙乌克兰印度尼西亚哈萨克斯坦塔吉克斯坦希腊南非蒙古奥地利肯尼亚加纳丹麦津巴布韦埃及坦桑尼亚捷克阿联酋安哥拉