yolov5第六版转为onnx,ncnn使用

20211118
yolov5第六版去掉了focus,在ncnn使用时兼容性更好。

固定形状输入转onnx

python export.py --include onnx --weight path/to/x.pt --img 640 --train --simplify

注意加–train,是去掉后处理部分的导出。

.

动态形状输入转onnx

python export.py --include onnx --weight path/to/x.pt --img 640 --train --dynamic --simplify

最后注意加–dynamic
待解决:simplify没有发挥作用?转ncnn失败

你可能感兴趣的:(计算机视觉,深度学习,yolo,onnx)