栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Java

YOLOv5

Java 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

YOLOv5

        本项目实现了Android设备读取USB摄像头并通过YOLOv5模型进行图像识别的功能,当然也支持本机摄像头。

·项目地址:

已分享至Github:https://github.com/10dutel/YOLOv5_Android_USBCamera

代码和文档可能会有更新。

·项目结构:

整个项目分为两个module:app和libusbcamera,其中libusbcamera为USB摄像头库,一般不修改。app模块集成了USB摄像头与移动版YOLOv5相关的Activity和工具类。

USB摄像头原项目地址:https://github.com/jiangdongguo/AndroidUSBCamera。注意项目中使用的是2.3.7版本的Demo,用Java编写。3.X之后的版本都是用Kotlin重构的。

移动版YOLOv5原项目:https://github.com/lp6m/yolov5s_android。注意这里的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 ————

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1031676.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号