【Android Studio】【NCNN】YOLOV8安卓部署

06-01 1891阅读

目录

下载Android Studio

克隆安卓项目 

关于自训练模型闪退问题 


下载Android Studio

下载Android Studio,配置安卓开发环境,这个过程比较漫长。

【Android Studio】【NCNN】YOLOV8安卓部署

安装cmake,注意安装的是cmake3.10版本。

【Android Studio】【NCNN】YOLOV8安卓部署

根据手机安卓版本选择相应的安卓版本,我的是红米K30Pro,安卓12。

【Android Studio】【NCNN】YOLOV8安卓部署

克隆安卓项目 

使用腾讯开源的ncnn,这是一个为手机端极致优化的高性能神经网络前向计算框架,能够将深度学习算法轻松移植到手机端高效执行。

克隆大佬写好的yolov8的安卓项目

git clone https://github.com/FeiGeChuanShu/ncnn-android-yolov8.git

下载解压ncnn-20231027-android-vulkan。

https://github.com/Tencent/ncnn/releases/download/20231027/ncnn-20231027-android-vulkan.zip

把ncnn-20231027-android-vulkan里面的文件放到ncnn-android-yolov8的\app\src\main\jni目录下。

【Android Studio】【NCNN】YOLOV8安卓部署

下载opencv

Releases - OpenCV

【Android Studio】【NCNN】YOLOV8安卓部署

然后解压,把文件夹放到ncnn-android-yolov8的\app\src\main\jni目录下。 

【Android Studio】【NCNN】YOLOV8安卓部署

修改ncnn-android-yolov8\app\src\main\jni里面的CMakeLists.txt文件,就是我们刚刚下载的两个文件的路径

【Android Studio】【NCNN】YOLOV8安卓部署

修改依赖的gradle插件版本为7.2.0,为什么呢,因为这个版本亲测没有问题,其他的难说。

【Android Studio】【NCNN】YOLOV8安卓部署

然后是修改使用的gradle版本为7.4-all版本。

【Android Studio】【NCNN】YOLOV8安卓部署

重新sync项目

【Android Studio】【NCNN】YOLOV8安卓部署

连接手机,打开手机的开发者模式,打开USB调试,打开USB安装,等Android Studio检测到我的红米K30Pro,然后点击运行,这时app就可以安装到手机上了,可以正常识别。

【Android Studio】【NCNN】YOLOV8安卓部署

yolov8安卓部署

另外app默认先打开的是前置摄像头,通过分析代码,可以知道,将MainActivity.java的40的facing的初始值从0改成1可以让app默认先打开后置摄像头

【Android Studio】【NCNN】YOLOV8安卓部署

关于自训练模型闪退问题 

这个是我后来用yolov8模型量化后部署发现app会闪退的问题,网罗解决办法,最后搞定

就是在用yolov8的export导出之前需要修改两个文件

from ultralytics import YOLO
model = YOLO('yolov8n.pt')
model.export(format='onnx', optimize=True)

修改ultralytics/nn/modules/block.py中的class C2f的forward函数

【Android Studio】【NCNN】YOLOV8安卓部署

        x = self.cv1(x)
        x = [x, x[:, self.c, ...]]
        x.extend(m(x[-1]) for m in self.m)
        x.pop(1)
        return self.cv2(torch.cat(x, 1))

修改ultralytics/nn/modules/head.py中的class Detect的forward函数

【Android Studio】【NCNN】YOLOV8安卓部署

        pred=torch.cat([xi.view(shape[0],self.no,-1)for xi in x],2).permute(0,2,1)
        return pred

然后再导出就行了,app不会再闪退

免责声明:我们致力于保护作者版权,注重分享,被刊用文章因无法核实真实出处,未能及时与作者取得联系,或有版权异议的,请联系管理员,我们会立即处理! 部分文章是来自自研大数据AI进行生成,内容摘自(百度百科,百度知道,头条百科,中国民法典,刑法,牛津词典,新华词典,汉语词典,国家院校,科普平台)等数据,内容仅供学习参考,不准确地方联系删除处理! 图片声明:本站部分配图来自人工智能系统AI生成,觅知网授权图片,PxHere摄影无版权图库和百度,360,搜狗等多加搜索引擎自动关键词搜索配图,如有侵权的图片,请第一时间联系我们。

目录[+]

取消
微信二维码
微信二维码
支付宝二维码