基于 YOLOv5 的安全帽检测方法研究
DOI:
作者:
作者单位:

作者简介:

通讯作者:

基金项目:


Research on Safety Helmet Detection Method Based on YOLOv5
Author:
Affiliation:

Fund Project:

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

    目的 针对工业场所背景复杂导致安全帽的检测精度低、效果不佳等问题,提出了一种基于 YOLOv5 的智能 检测安全帽的方法。 方法 首先在原模型 YOLOv5 的骨干网络中增加注意力机制,增强对不同尺寸目标特征的提 取,使得网络将注意力聚焦在含有安全帽的区域,增强了网络对安全帽信息的提取,以此有效提取安全帽的特征信 息;在预测层使用 EIoU 损失函数,考虑宽和置信度的差异、高和置信度的差异,把纵横比拆开,以此改善样本不平 衡问题,提升收敛速度的同时提高了回归精度。 结果 根据实验结果,改进的算法平均精度达到了 94. 7%。 相比于 YOLOv5 算法平均检测精度提高了 2. 2%,相比于 YOLOv3 算法平均检测精度提高了 12. 6%,可以有效地检测安全 帽。 结论 在同样的背景环境下,改进的算法可以有效地检测出远距离的小目标,对于复杂背景信息的图片,也可以 准确地检测出目标。 改进的算法有效地改善了原算法中小目标漏检和误检情况,也提高了检测精度。

    Abstract:

    Objective In response to the issues of low detection accuracy and poor performance of safety helmet detection in complex industrial environments a method for intelligent detection of safety helmets based on YOLOv5 was proposed. Methods First an attention mechanism was incorporated into the backbone network of the original YOLOv5 model which enhanced the extraction of features for targets of different sizes. This modification directed the network?? s attention toward regions containing safety helmets thereby improving the network?? s ability to capture safety helmet information and effectively extract the corresponding features. In the prediction layer the EIoU loss function was employed which considered the differences in width and confidence as well as height and confidence while splitting the aspect ratio. This approach addressed the issue of sample imbalance accelerated convergence and enhanced regression accuracy. Results According to experimental results the improved algorithm achieved an average precision of 94. 7%. The improved algorithm improved the average detection accuracy by 2. 2% compared with the YOLOv5 algorithm and 12. 6% compared with the YOLOv3 algorithm effectively detecting safety helmets. Conclusion Under the same background environment the improved algorithm can effectively detect small targets at long distances and accurately detect targets in images with complex background information. The improved algorithm effectively addresses the issues of missed detection and false detection of small targets in the original algorithm and also improves detection accuracy.

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

张帅帅.基于 YOLOv5 的安全帽检测方法研究[J].重庆工商大学学报(自然科学版),2025,(1):42-47
ZHANG Shuaishuai. Research on Safety Helmet Detection Method Based on YOLOv5[J]. Journal of Chongqing Technology and Business University(Natural Science Edition),2025,(1):42-47

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-12-24
×
2024年《重庆工商大学学报(自然科学版)》影响因子显著提升