YOLOv7安卓部署 ncnn

这篇具有很好参考价值的文章主要介绍了YOLOv7安卓部署 ncnn。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。


前言

本文将讲述如何利用ncnn在安卓手机端部署YOLOv7,这里以YOLOv7-tiny为例进行讲解,YOLOv7按照步骤依次进行即可,同时对于Android Studio的安装不在本次教程之内


1、YOLOv7-tiny模型部署

标准YOLOv7-tiny部署指的是没有更改YOLOv7-tiny的网络模型架构,直接利用训练好的模型权重进行部署,下面分几步进行讲述:

1.1 获得所需模型文件

首先是将训练获得的权重文件.pt转为.onnx

python export.py --weights '你的权重文件' \
--iou-thres 这个自己设置 --conf-thres 自己设置 \
--device 0

这个导出有GPU就GPU到,然后–fp16和–int8看自己情况来吧,导出之后会在你的权重保存路径下生成.onnx的文件,保存这个文件。
将导出的文件传到此网站:convertmodel

yolov7安装安卓app,YOLO,android
保存这两个文件,后续要用到

1.2 克隆所需要的代码

首先是下载ncnn-android-yolov7代码,其次是ncnn,之后便是opencv-mobile

ncnn-android-yolov7代码下载:
yolov7安装安卓app,YOLO,android

ncnn代码下载:
yolov7安装安卓app,YOLO,android
yolov7安装安卓app,YOLO,android

opencv-mobile下载:
yolov7安装安卓app,YOLO,android

下载之后便得到了三个压缩包,将其解压到同一目录下,最好目录不要有中文,避免报错,如下:
yolov7安装安卓app,YOLO,android

然后将红箭头所指文件夹复制到ncnn-android-yolov7的这个目录下:D:\xx\ncnn-android-yolov7-master\app\src\main\jni
yolov7安装安卓app,YOLO,android
yolov7安装安卓app,YOLO,android

之后用Android Studio打开ncnn-android-yolov7项目,打开箭头所指文件,然后将我画红线位置改成你所复制的文件夹名字:

yolov7安装安卓app,YOLO,android

将第1节得到的.param.bin文件复制到asserts文件夹下

到这里,准备工作就全部做好了,下一步就开始改代码了。

1.3 修改代码

请用netron打开刚刚转换模型得到的.param文件

yolov7安装安卓app,YOLO,android

这里的images填入到的ex.input("xx", yy);中的xx,请注意,images只是我的.param文件下的名字,如果你的和我的不一样,请填你模型的。

找到第一个检测头的这个卷积,
yolov7安装安卓app,YOLO,android
yolov7安装安卓app,YOLO,android
将画框的output填到stride8,这里的名字是onnx::Reshape_588,当然,你们的名字肯定和我的不一样,这个没关系,填你们自己的名字即可。依葫芦画瓢,找到第二个检测头上面一个卷积,然后将名字填到stride16,之后填stride32
yolov7安装安卓app,YOLO,android
yolov7安装安卓app,YOLO,android

对于yolo.cpp,请转到Yolo::detect函数,然后将得到的名字,按照相应位置填进去,对我画框部分进行修改,还有啊,anchor是你自己训练时配置文件中所用的anchor,所以也记得要进行修改。

yolov7安装安卓app,YOLO,android
yolov7安装安卓app,YOLO,android
yolov7安装安卓app,YOLO,android

继续在yolo.cpp文件中,转到Yolo::draw,改成你自己的类别名称:

yolov7安装安卓app,YOLO,android
到此为止yolo.cpp修改完毕


转到yoloncnn.cpp,按框所示,改成你自己的.param文件的名字:

yolov7安装安卓app,YOLO,android


转到下面箭头所指文件,app_name是做成app后显示的名字,然后model_array下面的item要填你的.param文件的名字

yolov7安装安卓app,YOLO,android


到此为止,代码上的改动已经全部结束,之后还需要对.param文件进行修改,打开它,然后像箭头一样修改即可

yolov7安装安卓app,YOLO,android
到此为止,真的就全部修改完了。

2、连接手机进行部署

首先打开开发者模式,这里不进行赘述,另外,如果你是小米手机,那么请在开发者选项中关闭MIUI优化,不然会出错
第一步:
yolov7安装安卓app,YOLO,android

第二步:
yolov7安装安卓app,YOLO,android

第三步:
yolov7安装安卓app,YOLO,android
静等一会儿,应用就会自动安装到手机,然后进行检测,到此为止,YOLOv7-tiny的部署就此完成。

总结

这里的图片和步骤有点多,请大家按照步骤一步一步进行,我也是看了一些资料才搞定的,应该算是保姆级教学了,原创,给个免费的赞谢谢!有问题可以在评论区交流哦!

参考

https://github.com/xiang-wuu/ncnn-android-yolov7
https://github.com/nihui/opencv-mobile
https://github.com/Tencent/ncnn文章来源地址https://www.toymoban.com/news/detail-772313.html

到了这里,关于YOLOv7安卓部署 ncnn的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包