树莓派上安装nanodet NCNN模型

本文详细介绍了如何在PC端搭建并训练NanoDet目标检测模型,随后将其转化为NCNN格式,并在树莓派上进行推理的过程。包括PyTorch到ONNX再到NCNN的转换步骤,以及树莓派上必要的环境配置。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

记录自己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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值