yolov5 pt权重文件转为ncnn格式

之前转换过一次，但是没有记录。导致这次转换的时候竟然记不清了。所以随手记录一下。

利用yolov5自带的export.py文件将pt文件转为onnx

python models/export.py --weights ****

***代表要转换的权重路径

 

之后会得到转换成功的onnx文件

 之后需要对**.onnx文件进行进一步的简化操作。

首先安装第三方包

pip install onnx-simplifier

之后：

python -m onnxsim best.onnx best-sim.onnx

得到简化后的best-sim.onnx文件

之后就可以进一步转化为ncnn格式了。这里使用一个大神的在线转换神器。

 一键转换 Caffe, ONNX, TensorFlow 到 NCNN, MNN, Tengine (convertmodel.com)

选择自己的模型进行转换，可能会出现报错

 遇到这种情况不要慌，先将转换出来的文件下载

 之后直接在yolov5打开.param文件。

将下面的结构

 修改为

 将原来的十层修改为了一层，所以185变成了176

之后将后面三层reshape修改为-1

 这样模型转换就搞定了，之后就可以去部署了。