美文网首页
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