美文网首页
ONNX: export failure 0.0s: Unsu

ONNX: export failure 0.0s: Unsu

作者: LabVIEW_Python | 来源:发表于2023-01-25 14:54 被阅读0次

问题 导出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模型

相关文章

网友评论

      本文标题:ONNX: export failure 0.0s: Unsu

      本文链接:https://www.haomeiwen.com/subject/sendhdtx.html