当前位置:   article > 正文

YOLOv5_Android_USBCamera:支持USB摄像头的YOLOv5Android图像识别项目_com.example.tflite_yolov5_test

com.example.tflite_yolov5_test

        本项目实现了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 ————

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Cpp五条/article/detail/263007?site
推荐阅读
相关标签
  

闽ICP备14008679号