think_deeply

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::

 

训练模型

1. U的YOLOV5正常训练模型,导出.pt文件

2. 将.pt文件转换成onnx,再转换成onnx-sim

python models/export.py --weights yolov5s.pt --img 640 --batch 1 --train #务必带上--train,否者后面使用onnx2ncnn会报错
python -m onnxsim yolov5s.onnx yolov5s-sim.onnx

 


安装ncnn

根据教程安装,同时参考下面两个链接:

1. https://www.jb51.net/article/183140.htm (不用单独装cmake)

2. https://github.com/Tencent/ncnn/wiki/how-to-build#build-for-windows-x64-using-visual-studio-community-2017 (mkdir -p在windows下不能运行,直接创建文件夹即可,例如build-vs2017)

3. 安装vs2017时要选择下面四个选项,否者报错。

 

 

 4. 还可以对文件做一次量化

ncnnoptimize yolov5s.param yolov5s.bin yolov5s-opt.param yolov5s-opt.bin 65536

 


 

坑:

1. 使用ncnn release的full-source编译(如ncnn-20220216-full-source.zip),而不是克隆或者下载ncnn的压缩包。

2. Build ncnn library时,先下载opencv包(https://github.com/opencv/opencv/releases/tag/4.5.5),并添加到cmake路径:

https://github.com/Tencent/ncnn/issues/1982

 

 

 

 

nihui大神:https://zhuanlan.zhihu.com/p/275989233

 

posted on 2022-03-01 21:48  think_deeply  阅读(522)  评论(0编辑  收藏  举报