Android实现车辆检测(含Android源码 可实时运行)

目录

Android实现车辆检测(含Android源码 可实时运行)

1. 前言

2. 车辆检测数据集说明

3. 基于YOLOv5的车辆检测模型训练

4.车辆检测模型Android部署

（1） 将Pytorch模型转换ONNX模型

（2） 将ONNX模型转换为TNN模型

（3） Android端上部署模型

（4） 一些异常错误解决方法

5. 车辆检测效果

6.项目源码下载

---

1. 前言

本篇博客是《YOLOv5实现车辆检测(含车辆检测数据集+训练代码)》续作Android篇，主要分享将Python训练后的YOLOv5的车辆检测模型移植到Android平台。我们将开发一个简易的、可实时运行的车辆检测Android Demo。

考虑到原始YOLOv5的模型计算量比较大，鄙人在YOLOv5s基础上，开发了一个非常轻量级的的车辆检测模型yolov5s05_320。从效果来看，Android车辆检测模型的检测效果还是可以的，高精度版本YOLOv5s平均精度平均值mAP_0.5=0.57192，而轻量化版本yolov5s05_416平均精度平均值mAP_0.5=0.47022左右。APP在普通Android手机上可以达到实时的检测识别效果，CPU(4线程)约30ms左右，GPU约25ms左右 ，基本满足业务的性能需求。

先展示一下Android Demo效果：

【Android APP体验】https://download.csdn.net/download/guyuealian/87244599

【项目源码下载】 Android实现车辆检测(含Android源码 可实时运行)

【尊重原创，转载请注明出处】https://panjinquan.blog.csdn.net/article/details/128190532

---

更多项目《智能驾驶 车牌检测和识别》系列文章请参考：

1. 智能驾驶 车牌检测和识别（一）《CCPD车牌数据集》：https://blog.csdn.net/guyuealian/article/details/128704181
2. 智能驾驶 车牌检测和识别（二）《YOLOv5实现车牌检测（含车牌检测数据集和训练代码）》：https://blog.csdn.net/guyuealian/article/details/128704068
3. 智能驾驶 车牌检测和识别（三）《CRNN和LPRNet实现车牌识别（含车牌识别数据集和训练代码）》：https://blog.csdn.net/guyuealian/article/details/128704209
4. 智能驾驶 车牌检测和识别（四）《Android实现车牌检测和识别（可实时车牌识别）》：https://blog.csdn.net/guyuealian/article/details/128704242
5. 智能驾驶 车牌检测和识别（五）《C++实现车牌检测和识别（可实时车牌识别）》：https://blog.csdn.net/guyuealian/article/details/128704276
6. 智能驾驶 红绿灯检测（一）《红绿灯(交通信号灯)数据集》：https://blog.csdn.net/guyuealian/article/details/128222850
7. 智能驾驶 红绿灯检测（二）《YOLOv5实现红绿灯检测(含红绿灯数据集+训练代码)》：https://blog.csdn.net/guyuealian/article/details/128240198
8. 智能驾驶 红绿灯检测（三）《Android实现红绿灯检测(含Android源码 可实时运行)》：https://blog.csdn.net/guyuealian/article/details/128240334

9. 智能驾驶 车辆检测（一）《UA-DETRAC BITVehicle车辆检测数据集》：https://blog.csdn.net/guyuealian/article/details/127907325

10. 智能驾驶 车辆检测（二）《YOLOv5实现车辆检测(含车辆检测数据集+训练代码)》：https://blog.csdn.net/guyuealian/article/details/128099672

11. 智能驾驶 车辆检测（三）《Android实现车辆检测(含Android源码 可实时运行)》：https://blog.csdn.net/guyuealian/article/details/128190532

---

2. 车辆检测数据集说明

目前收集了约10W+的车辆检测数据集：UA-DETRAC车辆检测数据集+Vehicle-Dataset车辆检测数据集+BITVehicle车辆检测数据集 

关于车辆检测数据集使用说明和下载，详见另一篇博客说明：《UA-DETRAC BITVehicle车辆检测数据集(含下载地址)》

---

3. 基于YOLOv5的车辆检测模型训练

官方YOLOv5给出了YOLOv5l,YOLOv5m,YOLOv5s等模型。考虑到手机端CPU/GPU性能比较弱鸡，直接部署yolov5s运行速度十分慢。所以本人在yolov5s基础上进行模型轻量化处理，即将yolov5s的模型的channels通道数全部都减少一半，并且模型输入由原来的640×640降低到416×416或者320×320，该轻量化的模型我称之为yolov5s05。从性能来看，yolov5s05比yolov5s快5多倍，而mAP下降了10%（0.57→0.47），对于手机端，这精度勉强可以接受。

下面是yolov5s05和yolov5s的参数量和计算量对比：

模型	input-size	params(M)	GFLOPs	mAP0.5
yolov5s	640×640	7.2	16.5	0.57192
yolov5s05	416×416	1.7	1.8	0.47022
yolov5s05	320×320	1.7	1.1	0.44788

yolov5s05和yolov5s训练过程完全一直，仅仅是配置文件不一样而已；碍于篇幅，本篇博客不在赘述，详细训练过程请参考：YOLOv5实现车辆检测(含车辆检测数据集+训练代码)

---

4.车辆检测模型Android部署

（1） 将Pytorch模型转换ONNX模型

训练好yolov5s05或者yolov5s模型后，你需要将模型转换为ONNX模型，并使用onnx-simplifier简化网络结构

# 转换yolov5s05模型
python export.py --weights "runs/yolov5s05_320/weights/best.pt" --img-size 320 320# 转换yolov5s模型
python export.py --weights "runs/yolov5s_640/weights/best.pt" --img-size 640 640

GitHub： https://github.com/daquexian/onnx-simplifier
Install:  pip3 install onnx-simplifier

（2） 将ONNX模型转换为TNN模型

目前CNN模型有多种部署方式，可以采用TNN，MNN,NCNN，以及TensorRT等部署工具，鄙人采用TNN进行Android端上部署：

TNN转换工具：

• （1）将ONNX模型转换为TNN模型，请参考TNN官方说明：TNN/onnx2tnn.md at master · Tencent/TNN · GitHub
• （2）一键转换，懒人必备：一键转换 Caffe, ONNX, TensorFlow 到 NCNN, MNN, Tengine   (可能存在版本问题，这个工具转换的TNN模型可能不兼容，建议还是自己build源码进行转换，2022年9约25日测试可用)

​

（3） Android端上部署模型

项目实现了Android版本的车辆检测Demo，部署框架采用TNN，支持多线程CPU和GPU加速推理，在普通手机上可以实时处理。Android源码核心算法YOLOv5部分均采用C++实现，上层通过JNI接口调用

package com.cv.tnn.model;import android.graphics.Bitmap;public class Detector {static {System.loadLibrary("tnn_wrapper");}/**** 初始化模型* @param model： TNN *.tnnmodel文件文件名（含后缀名）* @param root：模型文件的根目录，放在assets文件夹下* @param model_type：模型类型* @param num_thread：开启线程数* @param useGPU：关键点的置信度，小于值的坐标会置-1*/public static native void init(String model, String root, int model_type, int num_thread, boolean useGPU);/**** 检测* @param bitmap 图像（bitmap），ARGB_8888格式* @param score_thresh：置信度阈值* @param iou_thresh：  IOU阈值* @return*/public static native FrameInfo[] detect(Bitmap bitmap, float score_thresh, float iou_thresh);
}

如果你想在这个Android Demo部署你自己训练的YOLOv5模型，你可将训练好的Pytorch模型转换ONNX ，再转换成TNN模型，然后把TNN模型代替你模型即可。

（4） 一些异常错误解决方法

• TNN推理时出现：Permute param got wrong size

官方YOLOv5:  GitHub - ultralytics/yolov5: YOLOv5

Android实现车辆检测(含Android源码 可实时运行)相关推荐

1. Android实现红绿灯检测(含Android源码 可实时运行)

   Android实现红绿灯检测(含Android源码 可实时运行) 目录 Android实现红绿灯检测(含Android源码 可实时运行) 1. 前言 2. 红绿灯检测数据集说明 3. 基于YOLOv5 ...

2. Android Input子系统-含实例源码

   Android Input子系统-含实例源码 1 Input子系统作用 Android很多外设都是用到输入输出设备,比如touchscreen,键盘,音量键等,输入 设备对应Android 框架是An ...

3. Android实现手部检测和手势识别（可实时运行，含Android源码）

   Android实现手部检测和手势识别(可实时运行,含Android源码) 目录 Android实现手部检测和手势识别(可实时运行,含Android源码) 1. 前言 2. 手势识别的方法 (1)基于多 ...

4. 最全的Android开源音乐播放器源码汇总

   收集了很多音乐播放器类的Android项目源码,非常不错的开源项目,会让你事半功倍,希望大家补充...谢谢! Android基于经纬度切歌的冲绳音乐播放器源码 http://neast.cn/foru ...

5. [附源码]计算机毕业设计Python+uniapp基于Android的懂球帝APPqkwan(程序+源码+LW+远程部署)

   [附源码]计算机毕业设计Python+uniapp基于Android的懂球帝APPqkwan(程序+源码+LW+远程部署) 该项目含有源码.文档.程序.数据库.配套开发软件.软件安装教程 项目运行环境 ...

6. linux 浏览器源码下载,Android开发：图片浏览器源码

   Android 图片浏览器源码,是你学习Android很好的例子,希望对你的学习有帮助. main.xml android:layout_width="fill_parent" a ...

7. Android Jetpack 组件之 Lifecycle源码

   1.前言 最近简单看了下google推出的框架Jetpack,感觉此框架的内容可以对平时的开发有很大的帮助,也可以解决很多开发中的问题,对代码的逻辑和UI界面实现深层解耦,打造数据驱动型UI界面. A ...

8. Android多线程之ArrayBlockingQueue源码解析

   阻塞队列系列 Android多线程之LinkedBlockingQueue源码解析 Android多线程之SynchronousQueue源码解析 Andorid多线程之DelayQueue源码分析 ...

9. android注入 定位,[原创]修改源码实现全局(无需root)注入躲开注入检测

   看这篇文章需要的技能 1.会编译android源码(如果你不愿意编译源码,还有另外一种办法,下面我会提供) 2.会使用substrate或者xposed 以上2个网上资料很多我就不啰嗦了 一.市面上目 ...

最新文章

1. 使用母版页后出现控件，使用FindControl找不到指定控件
2. Shell：传递命令行参数
3. Redis专题-持久化方式
4. .NET Core 开源项目 Anet 在路上
5. c#+handle.exe实现升级程序在运行时自动解除文件被占用的问题
6. virtualbox 该内存不能为written_系统提示“该内存不能为read”的原因和解决办法...
7. python类和oop基础知识
8. 第一章 时间序列基础知识
9. 软件开发报价计算方式
10. 利用oc门或od门实现线与_OC与OD门.doc
11. Spring的两种定时器
12. Samsung Pay和Finablr宣布跨境支付合作
13. dns配置异常怎么修复_DNS错误怎么办（如何正确设置DNS）
14. 苹果手机用计算机打不开,苹果手机更新ios11后，照片在电脑上打不开怎么办？...
15. 即将到来的Xcode8 都更新了什么？
16. 打开资源管理器显示该文件没有与之关联的程序来执行该操作......
17. iOS Presenting view controllers on detached view controllers is discouraged
18. python的分布式爬虫框架
19. [blog] Speech Recognition Is Not Solved 语音识别领域尚待解决的子问题
20. Matlab 版本对照

热门文章

1. PS2游戏机硬盘启动制作教程
2. nohup 执行mysql命令_Linux nohup命令：后台命令脱离终端运行
3. I-Deas TMG 培训资料 (10)
4. JMeter 接口测试-if控制器
5. panda3d python_安装 Panda3D 并使用原有的Python
6. 计算机博弈六子棋估值函数,机器博弈中搜索策略和估值函数的设计
7. 如何将视频中的某一段截取制作gif动图
8. hudson安装以及使用
9. Uploadify/uploadifive上传(中文文档)
10. [Scene Graph] Neural Motifs: Scene Graph Parsing with Global Context 论文解读