YoloV5 Android APP开发--模型转换(三)

onnx和ncnn安装后，即可将模型转换，pt转onnx，再onnx转bin、param

导出模型，*.pt 转*.onnx

export PYTHONPATH="$PWD"

python export.py --weights yolov5l.pt --img 640 --batch 1

python export.py --train --weights yolov5s.pt --img 640 --batch 1

加—train参数防止 unsupport slice axes error

5l和5s均可用。

用5l 模型时注意要修改python源码。

模型优化

 python3 -m onnxsim yolov5s.onnx yolov5s-sim.onnx

.onnx转.bin和.param

cd ncnn-> build-> tools-> onnx

./onnx2ncnn yolov5ssim.onnx yolov5ssim.param yolov5ssim.bin

./onnx2ncnn yolov5ssim.onnx yolov5ssim.param yolov5ssim.bin

 可能会出现unsupported slice step！与yolov5模型输入层结构有关。