基于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 相关配置文件,本次实验选
- 配套讲稿:
如PPT文件的首页显示word图标,表示该PPT已包含配套word讲稿。双击word图标可打开word文档。
- 特殊限制:
部分文档作品中含有的国旗、国徽等图片,仅作为作品整体效果示例展示,禁止商用。设计者仅对作品中独创性部分享有著作权。
- 关 键 词:
- 基于 YOLOv4 算法 骑乘 人员 头盔 佩戴 远程 检测
1、咨信平台为文档C2C交易模式,即用户上传的文档直接被用户下载,收益归上传人(含作者)所有;本站仅是提供信息存储空间和展示预览,仅对用户上传内容的表现方式做保护处理,对上载内容不做任何修改或编辑。所展示的作品文档包括内容和图片全部来源于网络用户和作者上传投稿,我们不确定上传用户享有完全著作权,根据《信息网络传播权保护条例》,如果侵犯了您的版权、权益或隐私,请联系我们,核实后会尽快下架及时删除,并可随时和客服了解处理情况,尊重保护知识产权我们共同努力。
2、文档的总页数、文档格式和文档大小以系统显示为准(内容中显示的页数不一定正确),网站客服只以系统显示的页数、文件格式、文档大小作为仲裁依据,平台无法对文档的真实性、完整性、权威性、准确性、专业性及其观点立场做任何保证或承诺,下载前须认真查看,确认无误后再购买,务必慎重购买;若有违法违纪将进行移交司法处理,若涉侵权平台将进行基本处罚并下架。
3、本站所有内容均由用户上传,付费前请自行鉴别,如您付费,意味着您已接受本站规则且自行承担风险,本站不进行额外附加服务,虚拟产品一经售出概不退款(未进行购买下载可退充值款),文档一经付费(服务费)、不意味着购买了该文档的版权,仅供个人/单位学习、研究之用,不得用于商业用途,未经授权,严禁复制、发行、汇编、翻译或者网络传播等,侵权必究。
4、如你看到网页展示的文档有www.zixin.com.cn水印,是因预览和防盗链等技术需要对页面进行转换压缩成图而已,我们并不对上传的文档进行任何编辑或修改,文档下载后都不会有水印标识(原文档上传前个别存留的除外),下载后原文更清晰;试题试卷类文档,如果标题没有明确说明有答案则都视为没有答案,请知晓;PPT和DOC文档可被视为“模板”,允许上传人保留章节、目录结构的情况下删减部份的内容;PDF文档不管是原文档转换或图片扫描而得,本站不作要求视为允许,下载前自行私信或留言给上传者【自信****多点】。
5、本文档所展示的图片、画像、字体、音乐的版权可能需版权方额外授权,请谨慎使用;网站提供的党政主题相关内容(国旗、国徽、党徽--等)目的在于配合国家政策宣传,仅限个人学习分享使用,禁止用于任何广告和商用目的。
6、文档遇到问题,请及时私信或留言给本站上传会员【自信****多点】,需本站解决可联系【 微信客服】、【 QQ客服】,若有其他问题请点击或扫码反馈【 服务填表】;文档侵犯商业秘密、侵犯著作权、侵犯人身权等,请点击“【 版权申诉】”(推荐),意见反馈和侵权处理邮箱:1219186828@qq.com;也可以拔打客服电话:4008-655-100;投诉/维权电话:4009-655-100。