ONNX: export failure 0.0s: Unsu

2023-01-25  本文已影响0人  LabVIEW_Python

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

Unsupported ONNX opset version: 17

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

修改default为“11”

修改完毕后,再运行:

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

成功导出YOLOv5 onnx模型

上一篇下一篇

猜你喜欢

热点阅读