针对这个问题也翻了很多资料,我的解决方案如下:
参考:https://www.jianshu.com/p/cdff76fe7664
先说一下我的:pytorch版本:1.12.1 torchvision版本:0.13.1,onnx版本是1.13.0
然后是对yolov5中的export.py文件进行修改:
可以检索opset变量,文件中默认的是default=17,我只是对这句话中的opset默认值进行了修改,改成了我的onnx所对应的版本。
我看在上面定义的opset数值是12,但是我没有做修改,因为不太清楚原代码。
结果是生成了onnx文件,但是在编译过程中有许多的warring。
供大家参考下。