ONNX: export failure 0.0s: Unsupported ONNX opset version: 17

问题 导出YOLOv5 onnx模型时,报错!
版本:torch = 1.8.2; onnx = 1.13.0

解决方案 将export.py中的onnx opset 的默认值(default = 17)从“17”改为“11”

修改完毕后,再运行:

python export.py --weights yolov5s-seg.pt --include onnx

成功导出YOLOv5 onnx模型

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章