• 面向复杂施工现场的安全帽佩戴检测算法

    Safety helmet wearing detection algorithm for complex construction sites

    • 准确、实时检测安全帽佩戴是降低施工安全隐患的关键,但施工现场环境复杂,存在如人员密集、目标被遮挡、背景杂乱等问题,现有的安全帽检测算法难以适应,对密集小目标、被遮挡目标存在误检和漏检,且对算力要求较高。鉴于此,提出了一种基于YOLOv7-tiny的安全帽检测算法—DS-YOLO。在主干网络中,使用结合分布偏移卷积的DS-ELAN网络,并引入轻量化注意力机制,降低了浮点运算量,增强了对关键特征的提取能力;颈部网络中,通过结合了小目标检测层的BiFPN,加强了模型的多尺度特征融合能力,从而提升模型对小目标和密集目标的检测性能;使用WIoU Loss作为边界框回归损失函数,聚焦于普通质量的锚框从而提高模型性能。实验结果显示,DS-YOLO相较于YOLOv7-tiny浮点运算量下降了10.6%,所有目标场景下mAP提升了4.1%,小目标场景下mAP提高了3.2%,实现了36.6 frame/s的检测速度,模型在速度和精度上取得较好的平衡,更适合在算力不充足的真实施工现场环境中部署与应用。

       

      Abstract: Accurate and real-time detection of safety helmet wearing is the key to reducing construction safety hazards. However, the construction site environment is complex, with problems such as dense personnel, obscured targets, and cluttered backgrounds. Existing safety helmet detection algorithms are difficult to adapt to, and there are false detections and missed detections of dense small targets and obscured targets, and high computing power requirements are required. In view of this, a safety helmet detection algorithm based on YOLOv7-tiny is proposed, that is DS-YOLO. In the backbone network, the DS-ELAN network combined with distribution offset convolution is used, and a lightweight attention mechanism is introduced to reduce the amount of floating-point operations and enhance the ability to extract key features; in the neck network, the multi-scale feature fusion ability of the model is enhanced by combining BiFPN with a small target detection layer, thereby improving the model's detection performance for small and dense targets; WIoU Loss is used as the bounding box regression loss function to focus on anchor boxes of normal quality to improve model performance. Experimental results show that the floating-point computing amount of DS-YOLO is reduced by 10.6% compared with YOLOv7-tiny, the mAP in all target scenarios is increased by 4.1%, and the mAP in small target scenarios is increased by 3.2%. The detection speed of 36.6 frame/s is achieved. The model achieves a good balance between speed and accuracy, and is more suitable for deployment and application in real construction site environments with insufficient computing power.

       

    /

    返回文章
    返回