在Android端部署yolov5

闲来无事，记录一下自己在安卓端部署yolov5的步骤，历时一天

不想部署想直接拿来玩玩的，可以直接下载手机软件：https://download.csdn.net/download/qq_45077760/87629448

1.pt文件转onnx 

我用的是自己训练的权重文件，版本是6.1版本，直接执行以下代码对我的权重文件best.pt进行转化。过程可能会报错，如果报错就安装相应的库文件，一般是"no moudle named onnx",直接pip install onnx==1.7.0；因为如果直接装onnx，可能还是会报错，所以直接装1.7.0的onnx

python export.py --weights best.pt

转换之前记住一个细节问题！！！记得把train这里加上True。一定要注意，要不下边步骤会报错

执行完以上步骤，你会得到对应的best.onnx文件，这里我把名字best.onnx直接改成yolov5s.onnx，方便讲解

2.onnx文件转bin和param文件

这里网上有很多方法，但是都需要配置环境，这里要感谢一个github博主，他直接做了一个转换界面，太厉害了，网址附上：https://convertmodel.com/

点击进入界面，把你的onnx文件放入，记得步骤1转换时候加上 default=True，要不这里转换会报错，成功转换界面如下，如果出现红色就代表转换失败

 转换后会生成两个文件，点击下载生成的两个文件：yolov5s-slm-opt.param和yolov5s-slm-opt.bin

继续把yolov5s-slm-opt.param和yolov5s-slm-opt.bin名字改成yolov5s.param和yolov5s.bin

（ 注：如果你实在想自己走一遍流程，推荐两个博主的步骤，一个Ubuntu系统，一个win系统：

1.Ubuntu系统：https://blog.csdn.net/qq_44696500/article/details/124195375

2.win系统：https://blog.csdn.net/qq_28664681/article/details/112650644  ）

3.下载ncnn-android-yolov5和ncnn-20230223-android-vulkan

（1） 下载ncnn-android-yolov5文件，下载链接：https://github.com/nihui/ncnn-android-yolov5

解压后得到名为ncnn-android-yolov5-master的文件夹

（2）下载ncnn-20230223-android-vulkan文件，下载链接：https://github.com/Tencent/ncnn/releases

 解压后得到把文件夹里的以下四个文件拖进C:\Users\hp\Desktop\ncnn-android-yolov5-master\app\src\main\jni\这个路径

 （3）把步骤2生成的两个文件yolov5st.param和yolov5s.bin拉进C:\Users\hp\Desktop\ncnn-android-yolov5-master\app\src\main\assets这个文件夹里

 4.安装Android studio

下载链接：https://developer.android.google.cn/studio/

具体安装步骤网上都有，这里就不再写了

下载后用Android studio 打开ncnn-android-yolov5-master这个文件，然后点击左上角File-settings

再点击Android SDK，选择你手机的安卓版本，博主版本是9.0

 继续在这个界面点击SDK Tools，博主画红色框的记得选一下，SDK、cmake和NDK一定要选择

 配置一下NDK，点击左上角File--Project Structure， 这一步可能你的NDK选择会报错，因为正常情况下时会自动选择的，这是因为NDK版本装得太高了，可以执行上边步骤，重新安装低版本的NDK，我这里安装的是21.0.6113669，如果不行就多试几个版本就可以

5.修改源码

（1）修改CMakeLists.txt

打开ncnn-android-yolov5-master\app\src\main\jni\CMakeLists.txt，把里边的代码

set(ncnn_DIR ${CMAKE_SOURCE_DIR}/ncnn-20201218-android-vulkan/${ANDROID_ABI}/lib/cmake/ncnn)

替换成下边代码

set(ncnn_DIR ${CMAKE_SOURCE_DIR}/${ANDROID_ABI}/lib/cmake/ncnn)

 （2）修改yolov5s.param

在yolov5s.param里搜索reshape，把reshape后边的数值全修改成0=-1，如果不修改的话，部署后会出现很多密密麻麻的小检测框

 （3）修改yolov5ncnn_jni.cpp

在yolov5s.param里搜索Permute，如图所示，记住最后两个Permute后边画框的数值，我这里是353和367

 打开yolov5ncnn_jni.cpp文件，把最后两个Permute的数值353和367分别写进下边图中

 修改yolov5ncnn_jni.cpp文件里的类别，改成你的全中所用的类别名称

 6.编译工程

（1）这里重复使用可能会混乱，如果文件夹里有build这个文件记得先删除，没有的话不用理会

 然后点击Build--Clean Project

（2）点击build gradle这个文件运行等待生成，我这里有两个build gradle文件，点击红色这个，记得不要点错了

 （3）生成project

 （4）连接手机，点击运行按钮将软件下载到手机

7.部署完成