本文记录了yolov5迁移转ncnn模型的详细过程
流程三步走:1、pytorch转换成onnx; 2、onnx转换成ncnn;3、在android部署ncnn。
在此感谢逝去的头发 o(╥﹏╥)o
一、pytorch -> onnx
1、 打开yolov5/models/export.py,修改红色框路径
转换后我们得到 yolov5s.onnx
2、去掉onnx模型的冗余维度
pip install onnx-simplifier
python -m onnxsim yolov5s.onnx yolov5s-sim.onnx
至此我们得到简化后的onnx模型文件:yolov5s-sim.onnx