本项目实现了Android设备读取USB摄像头并通过YOLOv5模型进行图像识别的功能,当然也支持本机摄像头。
如需更换模型,请参考:YOLOv5_Android_USBCamera:YOLOv5模型定制化及转换_斜杠花括号的博客-CSDN博客
· 项目地址:
已分享至Github:https://github.com/10dutel/YOLOv5_Android_USBCamera
代码和文档可能会有更新。
· 项目结构:
整个项目分为两个module:app和libusbcamera,其中libusbcamera为USB摄像头库,一般不修改。app模块集成了USB摄像头与移动版YOLOv5相关的Activity和工具类。
USB摄像头原项目地址:AndroidUSBCamera。注意项目中使用的是2.3.7版本的Demo,用Java编写。3.X之后的版本都是用Kotlin重构的。
移动版YOLOv5原项目:移动YOLO5。注意这里的targetSdkVersion使用的29(Android 10),但为配适libusbcamera,应当改为27,否则会引起底层的权限问题。本项目中已修改。
· 文件说明
app模块的目录结构如图
· com/jiangdg/usbcamera/view/USBCameraActivity.java
USBCamera和TfliteRunner检测器的初始化。TfliteRunner的参数配置在onCreate()里的STEP.0中实现。同样在onCreate()下对CameraHelper初始化,在mCameraHelper.setOnPreviewFrameListener()中实现回调方法onPreviewResult,利用单线程池执行检测任务并展示。
提供了centerSquareScaleBitmap方法,在执行检测前需要把输入图像裁剪为320*320的正方形(视模型而定)。
· com/example/tflite_yolov5_test/camera/DetectorActivity.java
本地摄像头Activity,USBCameraActivity中检测器的配置参考了这里的参数。
· com/example/tflite_yolov5_test/MainActivity.java
在原项目中有而此处未显示的界面,提供了模型的选择和输入本地图像检测的功能。如需添加模型切换的功能可以参考这里的实现。
· com/serenegiant/usb/common/AbstractUVCCameraHandler.java
其中包含一个CameraThread内部类用以具体执行Handler的消息,调用底层方法进行USB摄像头图像读取与处理的实现类。AbstractUVCCameraHandler本身负责多线程通信。
如需更换自己训练的模型请按yolov5s_android项目的方法将模型文件转换后放置于asset目录下。
———— 2022/7/28 end ————