改进YOLOv5s算法的电动车头盔检测研究

作     者：侯恩翔 张旭 刘罡 张秀再 Hou Enxiang;Zhang Xu;Liu Gang;Zhang Xiuzai

作者机构：南京信息工程大学电子信息工程学院南京210044 无锡学院电子信息工程学院无锡214105 

基　　金：国家自然科学基金(62204172)项目资助 

出 版 物：《国外电子测量技术》 (Foreign Electronic Measurement Technology)

年 卷 期：2024年第43卷第3期

页      码：168-176页

摘      要：针对电动车头盔佩戴检测存在遮挡、车辆密集以及一车多人的复杂场景下出现的漏检、误检问题,在YOLOv5s的基础上,提出了一种应用于电动车头盔佩戴检测的改进算法。设计了一种由递归门控卷积改进的GBC3模块,替换网络主干和特征融合层(feature pyramid networks,FPN)中的C3模块,加强邻间特征的空间信息交互,提高网络的特征提取和特征融合能力;其次在主干和特征融合网络添加无参注意力机制(SimAM),以调整特征图中不同区域的注意力权重,对重要目标施加更多关注;最后引入调整超参后的WIOU损失函数,优化预测框回归,提高模型的目标定位能力。在自制电动车头盔数据集上的实验结果表明,改进模型在仅增加较少参数的前提下,其平均精度均值(mAP)达到97.3%,较YOLOv5s提高了3.2%,并且检测速度为87.1 fps,改善了误检和漏检的问题,同时仍具有较高的实时性,更适用于电动车驾乘者的头盔佩戴检测。

主 题 词：电动车头盔检测 递归门控卷积 空间交互 注意力机制 

学科分类：1305[艺术学-设计学类] 13[艺术学] 08[工学] 081104[081104] 0804[工学-材料学] 081101[081101] 0811[工学-水利类] 

D　O　I：10.19652/j.cnki.femt.2305586

馆 藏 号：203127497...