![点击分享此内容 分享](/master/images/share_but.png)
基于YOLOv4算法的骑乘人员头盔佩戴的远程检测.pdf
《基于YOLOv4算法的骑乘人员头盔佩戴的远程检测.pdf》由会员分享,可在线阅读,更多相关《基于YOLOv4算法的骑乘人员头盔佩戴的远程检测.pdf(6页珍藏版)》请在文库网上搜索。
1、第 61 卷 第 10 期Vol.61 No.102023 年 10 月October 2023农业装备与车辆工程AGRICULTURAL EQUIPMENT&VEHICLE ENGINEERINGdoi:10.3969/j.issn.1673-3142.2023.10.035基于 YOLOv4 算法的骑乘人员头盔佩戴的远程检测李明,杜茂华(650500 云南省 昆明市 昆明理工大学 机电工程学院)摘要 针对现有头盔检测模型易将非骑行者纳入检测范围的错误识别问题,提出将人与车整体标定的方法来制作骑行者佩戴头盔数据集,然后采用 YOLOv4 算法进行模型训练,可以正确识别出骑行状态与非骑行状态。
2、通过二分K-means算法对拟识别目标先验框进行聚类分析,采用分步训练方式优化学习权重,识别精度达到了90.8%。其次,为了将交通执法人员从危险环境解放出来,提出网络摄像头截取视频流并通过以太网输入到 PC 端 YOLOv4 模型中的方法,替代人工实现远程检测。实验表明,该方案能正确识别出视频中骑行人员有无佩戴头盔,证明了该方案的有效性。关键词 摩托车头盔;远程监测;YOLOv4 算法;二分 K-means 算法;网络摄像头 中图分类号 TP391.41;U495 文献标志码 A 文章编号 1673-3142(2023)09-0159-06引用格式:李明,杜茂华.基于 YOLOv4 算法的骑乘
3、人员头盔佩戴的远程检测 J.农业装备与车辆工程,2023,61(10):159-164.Remote detection of helmets for riders based on YOLOv4LI Ming,DU Maohua(School of Mechanical and Electrical Engineering,Kunming University of Science and Technology,Kunming 650500,Yunnan,China)Abstract Traffic in cities is complex,which causes traffic acci
4、dents easy to happen.When motorcycle riders wear helmets during riding,their accidental injuries can be reduced to a certain extent.Therefore,it is of great significance to carry out remote detection on whether riders wear helmets or not during riding.Aiming at the erroneous identification problem t
5、hat exists in current helmet detection models which tend to include non-cyclists in the detection range,the model in this paper proposed the method of overall calibration of taking people and motorcycles as a whole to produce helmet wearing data sets of cyclists,and then YOLOv4 algorithm was used fo
6、r model training,so that the cycling state and non-cycling state could be correctly identified.The binary K-means algorithm was used to perform clustering analysis on the priori frame of the target to be recognized,and the learning weight was optimized by step training method,and the recognition acc
7、uracy reached 90.8%.Secondly,in order to liberate the traffic law enforcement personnel from the dangerous environment,a method was proposed to capture the video stream by webcam and input it to YOLOv4 model at PC terminal through Ethernet instead of manual so that remote detection could be achieved
8、.Experimental results showed that the scheme could correctly identify whether the cyclists were wearing helmets in the video,which proved the effectiveness of the scheme.Key words motorcycle helmet;remote monitoring;YOLOv4;binary K-means;webcam0 引言摩托车、电动车、自行车等骑乘交通工具以其便捷、便宜、环保等特点受到出行者的广泛欢迎,但这类骑乘人员在驾驶
9、过程中完全暴露于道路之中,容易在交通事故中伤亡。2019 年,我国共发生交通事故 244 058 起,其中摩托车、非机动车、自行车交通事故发生数共 76 967 起,占比约为 31.54%。摩托车、非机动车、自行车发生交通事故致死人数占总交通事故死亡人数的24.91%1。骑行头盔是骑乘交通工具不可缺少的一部分,可以有效减少发生交通事故时的伤亡。近几年全国各地陆续开始实施相关法规,要求骑乘者必须佩戴安全头盔。在各个交通要道安排执勤人员检测骑乘者头盔佩戴情况,但是这种检测方式效率低、漏检率高,对执勤人员的生命安全也存在一定威胁。近十年人工智能崛起,机器学习代替了人工学习,极大地提高了工作效率,尤其
10、在图像处理领域,深度学习成果显著。采用深度学习网络对骑行人员进行头盔佩戴检测能有效解决当前所存在的问题。目前,基于深度学习的目标检测算法主要分为收稿日期:2022-09-19160农业装备与车辆工程 2023 年2 类:(1)两阶段(Two-stage)算法,如 RCNN2(Region-proposal Convolutional Neural Network)、Fast R-CNN3、Faster R-CNN4、R-FCN5(Region-based fully convolutional network)、Mask R-CNN6等;(2)单阶段(One-stage)算法,如 SSD7(S
11、ingle Shot MultiBox Detector)、YOLO8(You Only Look Once)等。这 2 类算法的区别在于是否需要生成候选框。一般而言,使用两阶段算法的模型在预测精度上表现优秀,但 2 步计算方式增加了计算成本,影响网络训练速度。YOLO、SSD 等单阶段算法将目标边框定位问题转化为回归问题,可以进行端到端检测,因为没有候选框,所以速度快、实时性好,被广泛应用于工程实际中。头盔检测应用的场景是车水马龙的交通道路,环境复杂,且人流车流量大。不但要对庞大的车流人流进行分析,还要从中识别出骑行者有无佩戴头盔,所以对模型运算精度和实时性都有较高的要求。因此,本文采用具有
12、较高精度且时效性更好的单阶段算法 YOLOv4 模型。传统检测方法将待检测目标划分为 2 部分:骑车人和骑乘工具9-10,这样制作的数据集极易将非骑行人员纳入识别范围,增加了拟识别对象的数目,也就增加了计算量和计算过程,所以在人员密集环境下的检测精度不高。为此,本文采用人与车整体标定的方法,即在制作数据集时,将骑车人与骑乘工具标定为一个整体。这样,训练出的网络就可有效地将骑车者与非骑车者区别开来;同时,为了实现远程监控功能,采用以太网搭配网络摄像头的方式,将现场采集的视频流通过以太网实时输送到 YOLOv4 模型中进行分析预测,以实现远程目标的检测。1 YOLOv4 算法原理2020 年 4
13、月,Bochkovskiy 等11在 前 3 版YOLO 模型的基础上,提出了 YOLOv4 的网络模型,该 模 型 将 Cross-Stage-connection(CSP)、Self-adversarial-training(SAT)、Mosaic data augmentation 等技术融合,使 YOLOv4 性能得到大幅度提高。图 1 所示为各神经网络的性能对比。与 YOLOv3 相比 YOLOv4 网络主要有以下改进:(1)使用 CSPDarknet5312代替 Darknet53 作为特征提取网络,目的是解决复杂卷积神经网络框架主干中网络优化梯度信息重复的问题13。CSPDark
14、net53 模块示意图如图 2 所示;(2)使用SPP14网络作为 Neck 的附加模块,主要解决输入图像尺寸与所要求尺寸不匹配、对图像进行拉伸和裁剪时易失真的问题。该方法应用于 YOLOv4 中,增强了网络感受野。网络结构示意图如图 3 所示;(3)使用 PANet15作为 Neck 的特征融合模块,进一步加强了 YOLOv4 的特征提取能力。2 实验实验平台是 Windows10(64位)操作系统,CPU为 AMD Ryzen 7 4800H,2.9 GHz,GPU 为 NVIDIA GTX1650Ti,显存为 4 G。CUDA 版本为 10.2,CuDNN 版本为 7.6.5,采用的编程
15、语言为 Python3.7。2.1 数据集制作为了适应在复杂交通环境中对来往骑行车辆进行头盔检测,本文采用网络爬取图片加摄像头拍摄的方式制作数据集。网络爬取的目的是收集不同拍摄角度骑行状态的图片,便于后期模型能学到多种特征,进而提高检测的精度。摄像头拍摄马路上真实交通状态,一张图片中包含多个检测目标,有利于后期训练出的模型对密集目标的检测,更好地拟合实际交通情况。图片采集情况如图 4 所示。为了解决现有模型9错误地将行人判定为待检测目标的问题,本文采用整体标定的方式对拟检测目标进行标定,使得模型检测的范围锁定为骑行者而非其他目标,整体标定例图如图 5 所示。因为对目标进行检测属于监督学习的范畴
16、,所以要对训练集中待检测的目标进行位置和类别的标注。采用图 2 CSPDarknet53 模块示意图Fig.2 CSPDarknet53 module diagramCBM CSP1 CBM CSP2 CBM CSP3 CBM 2 2 2YOLOv4 BackBoneCBM CSP8 CBMCBM CSP4 CBM19196086083CBM=ConvBNMishResUnit=CBL CBLaddCSPxCBLCBLConcatCBL=ResUnitx个残差组件图 1 各神经网络的性能对比Fig.1 Performance comparison of each neural network1
17、0 30 50 70 90 110 130FPS(V100)5048464442403836343230APMS COCO Object DetectionEfficlentDetYOLOv4ASFF*YOLOv3ATSSCenterMask*161第 61 卷第 10 期Labeling 软件对图片进行逐一标注,生成 xml 文件,文件格式如图 6 所示。本次数据集一共采集 1 500张图片,其中 75%作为训练集,25%作为测试集。2.2 二分 K-means 先验框聚类分析对于一个给定的样本,样本里包含 n 个对象,X=X1,X2,X3,Xn,随机设置 K 个特征空间内的点作为初始的聚类
18、中心 m1,m2,m3,mk,然后计算每个点到 K 个中心的欧式距离,未知的点选择最近的一个聚类中心点作为标记类别。欧式距离表达式为,dis X mXmiiititm21=-=hh/(1)式中:Xi第 i 个对象;mi第 i 个聚类中心;Xit第 i 个对象的第 t 个属性。采用误差二次方和准则函数作为聚类准则函数,误差二次方和 J 为Jxmijijnik112i=-=h/(2)式中:ni第 i 个聚类样本点数量;mi第 i个聚类中心;Xij第 i个聚类中的第 j个样本点。本实验所涉及的对象只有戴头盔与未带头盔 2种,因此需要重新对先验框进行聚类分析。K-means算法具有收敛速度快、聚类效果
19、较好的优点,但得到的结果和运行时间容易受初始聚类中心的影响,一旦初始聚类中心选取不恰当,就容易造成局部最优解。为了解决局部最优解问题,本实验采用二李明 等:基于 YOLOv4 算法的骑乘人员头盔佩戴的远程检测图 3 YOLOv4 网络结构示意图Fig.3 Schematic diagram of YOLOv4 network structure图 4 图片采集Fig.4 Image acquisition图 6 xml 文件Fig.6 xml fileCBM CSP1 CBM CSP2 CBM CSP3 CBM 2 2 22.BackBone1.输入端CBM CSP8 CBMCBM CSP4
20、CBM CBLSSP CBL6086083CBM=ConvBNMishCBL=ConvBNLeaky ReLUResUnit=CBL CBLaddCSPxCBLCBLConcatCBL=ResUnitx个残差组件CBL 上采样CBLCBLConcatCBL 上采样ConcatConcatConcatCBLCBLCBLCBLCBLCBL 3 3 5 5 5 5CBL ConvCBL ConvCBL Conv3.Neck3.Prediction38 38 25576 76 25519 19 255SSPMaxpoolMaxpoolMaxpoolConcat图 5 整体标定示意图Fig.5 Over
21、all calibration diagram162农业装备与车辆工程 2023 年分 K-means 算法对先验框进行聚类分析。该算法不需要选择初始聚类中心,由一个样本集分裂后得到 2 个簇,对于 K 个簇则进行 K-1 次分裂,具体流程如图 7 所示16。实验得到在 K=9 时先验框最优,分别为(35,84)、(80,118)、(115,162)、(92,282)、(151,219)、(148,441)、(223,338)、(285,500)、(431,543)。之后再将新的聚类结果更新到 YOLOv4 的配置文件中进行训练。2.3 模型训练首先修改 YOLOv4 相关配置文件,本次实验选
- 1.请仔细阅读文档,确保文档完整性,对于不预览、不比对内容而直接下载带来的问题本站不予受理。
- 2.下载的文档,不会出现我们的网址水印。
- 3、该文档所得收入(下载+内容+预览)归上传者、原创作者;如果您是本文档原作者,请点此认领!既往收益都归您。
下载文档到电脑,查找使用更方便
10 文币 0人已下载
下载 | 加入VIP,免费下载 |
- 配套讲稿:
如PPT文件的首页显示word图标,表示该PPT已包含配套word讲稿。双击word图标可打开word文档。
- 特殊限制:
部分文档作品中含有的国旗、国徽等图片,仅作为作品整体效果示例展示,禁止商用。设计者仅对作品中独创性部分享有著作权。
- 关 键 词:
- 基于 YOLOv4 算法 骑乘 人员 头盔 佩戴 远程 检测
![提示](https://www.wenkunet.com/images/bang_tan.gif)