这里写目录标题
一、引言
想实现下在Android上部署个深度学习模型,本文可以同时部署5.0或6.0版本模型,正文如下:
二、准备工作
所有用到的软件都放在这个百度网盘里,[点击下载,链接:https://pan.baidu.com/s/1BXGknC0xTDCTx-FIiRG88Q
提取码:q7m8]
1、Android Studio
安装Android Studio,
2、安装vs2019
我采用的是2019版本,一开始用的2022版本出错了,因此降低了版本。软件下载在网盘里
下载VS后安装下面3个工作负荷
已经有VS的读者补充工具负荷即可:工具->获取工具和功能
3、ncnn-yolov5-android源码
onnx转ncnn模型
省事版:一键转换网站,onnx->ncnn
转换后会得到param和bin模型
三、模型部署
1、修改export文件,转化onnx模型
现有的export代码转化的onnx模型会带着一个Detect层(后处理层),与安卓推理所需要三层输出不对应,因此需要修改yolo.py文件里的Detect()函数,加上一个True。
运行后会报错,所以需要将export文件里的这两行注释掉,就可以运行了,出来的模型就带了三个输出。
2、 下载源码
3、转换模型
- 传统方法https://blog.csdn.net/CSDNXXXR/article/details/125502127
- 通过一键转换网站得到param和bin模型,将模型复制到代码里的assets文件里,
4. 模型修改及部署
将模型复制到这个目录下
(1) 如果是5.0版本,需要改Focus结构,
修改后
减去Focus结构的层数,
即181-6=175
(2)如果是6.0以上版本的模型,没有focus结构,只需要修改最后reshape的0=-1,
然后找到最后两行的Permute的输出值,一个是365,一个是385
将这两个值放入CPP文件里的stride16和stride32里,就万事大吉了!
然后点击运行runapp就可以了
5.Android Studio
sync生成配置环境
SDK配置环境
缺少文件的话
在ndk
错误1:apk安装失败-15: 解决方法
错误2:No toolchains found in the NDK toolchains folder for ABI with prefix: arm-linux-androideabi :解决方法
错误3: