yolov5模型转换

yolov5本身release目录有提供了onnx转换好的模型,想着也自己操作一遍,可是实际操作却遇到了问题,这里做下记录方便后续可能用到

yolov5模型转换_第1张图片

安装onnx,转的时候提示出错ONNX: export failure  0.1s: Unsupported ONNX opset version: 17

修改export.py里的默认设定如下,原先默认是17,改为12后可以正常转,会在目录生成转换的onnx模型

yolov5模型转换_第2张图片

验证转换后的onnx模型时会提示出错,按提示安装pip install onnxruntime

yolov5模型转换_第3张图片

进行验证

yolov5模型转换_第4张图片

yolov5模型转换_第5张图片

转换用的模型通过Netron 进行查看

yolov5模型转换_第6张图片

你可能感兴趣的:(AI,AI,yolov5)