yolov5s.pt转onnx报错：ONNX: export failure ❌ 0.0s: Unsupported ONNX opset version: 17

参考

1. https://github.com/Hexmagic/ONNX-yolov5/issues/13
2. https://github.com/ultralytics/yolov5/blob/ac6c4383bc0c7a2a4f7ca18f8733821b49e916bd/export.py#L908

描述

python export.py --weights yolov5s.pt --include onnx --device 0

运行该命令实现yolov5的export.py文件执行转换pt格式的yolo模型文件为onnx格式中报错ONNX: export failure ❌ 0.0s: Unsupported ONNX opset version: 17

解决方法

执行

python export.py --weights yolov5s.pt --include onnx --device 0 --opset 11

添加 --opset 11参数