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模型输入层结构有关。