torch-onnx-ncnn过程(yolov5)
torch-onnx-ncnn 以yolov5为例子安装ncnn如果需要用vscode进行代码调试torch2onnx下载测试转换onnxonnx2ncnn1.生成param和bin文件2.打开生成的.param文件,去除不支持的网络层。3.除去split 和crop 及concat层,一共十层,换成YoloV5Focus,总层数309变成300,如图所示。4.将Permute的前一层Reshape 中0=xxx 改成 0 = -1。修改结果如下:5. 用 ncnnoptimize 过一遍模型,顺便转为 f
原创
2021-10-20 15:03:04 ·
1494 阅读 ·
8 评论