记录自己nanodet 环境搭建 和运行的过程
首先在PC端搭建nanodet的训练模型根据GitHub上的安装流程进行安装,pytorch要选择gpu版本。
训练部分参考了这两篇博客 需要自己修改里面的一些东西
https://www.cnblogs.com/azureology/p/14103685.html
https://github.com/RangiLyu/nanodet
训练完成之后讲nanodet的pytorch模型转成ncnn模型加快模型的推理。首先是将.pth文件转成onnx 模型之后进行根据虾米那这个网址转成ncnn的.param文件和.bin文件
https://convertmodel.com/
转成之后再树莓派上进行推理
主要参考这个网址里面的过程
https://github.com/RangiLyu/nanodet/blob/main/demo_ncnn/README.md
首先是要安装protobuf
https://blog.youkuaiyun.com/qq_36780295/article/details/108486308
文章里的make 和make check 一定要加上sudo否则会失败
之后安装opencv
https://blog.youkuaiyun.com/qq_36486890/article/details/97511295
make的时间特别长需要耐心等待
最后安装ncnn 编译 还是根据这里安装
https://github.com/RangiLyu/nanodet/blob/main/demo_ncnn/README.md
安装之后一定要将ncnn放到环境变量中。
之后把自己训练的模型文件 放到 nanodet的demo中运行就可以了
https://github.com/RangiLyu/nanodet