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

文章预览

点击上方 蓝字 关注我们 微信公众号： OpenCV学堂 关注获取更多计算机视觉与深度学习知识 错误信息 昨天有个课程学员跟我反馈说YOLOv5导出的ONNX格式模型，OpenCV DNN无法读取的问题，我以为是它使用的OpenCV版本不是4.7以后的，当他告诉我使用的OpenCV DNN版本是4.10.x时候，我实在无法想出是什么问题，于是今天请他把模型文件跟代码都发我了，我直接运行了一下，发现的确会有读取错误，错误信息如下： 发现的确是无法加载，但是我用本地的我之前导出来的YOLOv5的模型，发现加载没有问题，没有错误，于是我意识到肯定导出的ONNX格式的模型文件有相关的差异，用Netron打开以后对比信息如下： 对比发现只有这两个地方不一样，查找官方资料说法如下： 意思是导出的模型不是JIT模式，但是我不知道main graph是怎么导出来的，是不是Pytorch2.4已经支持这种模 ………………………………