改进 YOLOv5s算法的电动车头盔检测研究
DOI:
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP391.4

基金项目:

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


Improved YOLOv5s algorithm for electric bike helmet wearing detection
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

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

    Abstract:

    Aiming at the leakage and misdetection problems of electric vehicle helmet wearing detection in the presence of occlusion,dense vehicles and the complex scene of multiple people in one vehicle,an improved algorithm applied to electric vehicle helmet wearing detection is proposed on the basis of YOLOv5s.A GBC3 module improved by recursive gating convolution is designed to replace the C3 module in the network backbone and feature fusion layer(feature pyramid networks,FPN),strengthen the spatial information interaction of neighbor features,and improve the feature extraction and feature fusion capabilities of the network.Secondly,non-parametric attention mechanism(a simple, parameter-free attention module for convolutional neural networks,SimAM)is added to the backbone and feature fusion network to adjust the attention weight of different regions in the feature map and pay more attention to important targets.Finally,the WIOU loss function is introduced to optimize the prediction box regression and improve the target localization ability of the model.The experimental results on the self-made electric vehicle helmet dataset show that the mAP of the improved model reaches 97.3%under the premise of only adding fewer parameters,which is 3.2%points higher than that of YOLOv5s,and the detection speed is 87.1 fps,which improves the problem of false detection and missed detection,and still has high real-time performance,which is more suitable for helmet wearing detection of electric vehicle drivers.

    参考文献
    相似文献
    引证文献
引用本文

侯恩翔,张 旭,刘 罡,张秀再.改进 YOLOv5s算法的电动车头盔检测研究[J].国外电子测量技术,2024,43(3):168-176

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-06-12
  • 出版日期:
文章二维码