本文档介绍了如何使用ONNX进行模型转换，将PyTorch的.pt模型转换为ONNX格式，然后进一步将ONNX模型优化并转换为NCNN的.bin和.param文件。过程中涉及的关键步骤包括导出模型、模型优化以及使用ONNX到NCNN的转换工具。注意在转换过程中可能遇到的‘unsupported slice step’错误，并提供了相应的解决建议。
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
最低0.47元/天 解锁文章
扫一扫
7408
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
