报错如下:

先说解决办法:将1.16.2版本的onnx降级为1.16.1

1)具体操作:

本人使用的是anaconda虚拟环境,故操作分两步:卸载旧版本onnx,安装指定版本

激活的虚拟环境中使用CLI命令 ,卸载原有版本1.16.2

pip uninstall onnx

然后使用命令安装onnx的1.16.1版本(其它低版本可能也行,本人没过多尝试)

pip install onnx==1.16.1

ok!此时运行命令,即可成功导出

 yolo export model=yolov8n.pt format=onnx simplify=True

导出结果图:

2)conda虚拟环境中主要的库版本

python                    3.8.0 

torch                     2.4.0+cu118 

torchaudio                2.4.0+cu118 

torchvision               0.19.0+cu118 

ultralytics               8.2.74 

onnxruntime               1.18.1 

onnx-simplifier           0.4.36

onnxruntime-gpu           1.18.1 

onnxsim                   0.4.36

onnxslim                  0.1.32

3)排错(踩坑)过程

重装ultralytics,重装onnx,onnxruntime,以及修改python版本等均没解决问题,之前以为是缺乏库或者依赖的问题,同样的环境更换为1.16.1版本onnx成功导出

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐