• 基于HFD-YOLO的安全帽检测算法

    Safety helmet detection algorithm based on HFD-YOLO

    • 为降低工人安全事故发生率并有效增强其安全意识,本文提出一种面向煤矿场景的安全帽检测算法HFD-YOLO。以YOLOv11n目标检测算法为基线,首先设计了C3k2-PIPA结构,增强模型对安全帽的特征提取能力,以解决安全帽目标小、特征易受干扰等问题;其次,设计了CCCM-Concat结构,用于融合深层与浅层特征信息,实现深层与浅层之间的高效特征交互,利用局部特征补充全局信息,从而避免因网络加深导致的信息丢失;最后,设计了MCIM-P2Detect结构,提升模型对多尺度特征的检测能力,使其能够更好地适应不同尺度变换下的目标检测需求。实验结果表明,改进后模型的mAP@0.5较基线模型提升4.8%,达到80.2%,能够满足工人安全帽检测的实际应用要求。

       

      Abstract: To reduce the incidence of worker safety accidents and effectively enhance safety awareness, this paper proposes a helmet detection algorithm for coal mine scenes, named HFD-YOLO. Using the YOLOv11n object detection algorithm as the baseline, the following improvements are introduced. First, a C3k2-PIPA structure is designed to enhance the model's feature extraction capability for helmets, addressing the challenges of small helmet targets and easily disturbed features. Second, a CCCM-Concat structure is designed to fuse deep and shallow feature information, enabling efficient feature interaction between deep and shallow layers. Local features are used to complement global information, thereby avoiding information loss caused by network deepening. Finally, an MCIM-P2Detect structure is designed to improve the model's multi-scale feature detection capability, enabling it to better adapt to object detection under varying scale transformations. Experimental results demonstrate that the improved model achieves a 4.8% increase in mAP@0.5 compared to the baseline model, reaching 80.2%, which meets the practical application requirements for worker helmet detection.

       

    /

    返回文章
    返回