基于改进YOLO11n的矿井人员安全帽佩戴检测算法

    Mine personnel helmet wearing detection algorithm based on improved YOLO11n

    • 摘要: 针对矿井复杂环境下工作人员安全帽小目标检测算法精度低、模型复杂度高等问题,设计了一种基于YOLO11n的改进检测算法。首先,设计了跨阶段部分多尺度特征聚合CSP−PMSFA(Cross Stage Partial−Partial Multi−Scale Feature Aggregation)模块,该模块采用3×3、5×5和7×7 3种不同尺寸的卷积核并行提取特征,通过部分选择性连接机制实现了多尺度特征有效融合,增强多尺度安全帽细节特征的捕捉能力;其次,在Neck层设计了小目标增强金字塔SOEP(Small Object Enhance Pyramid)模块,结合SPDConv(Space−to−Depth Convolution)空间特征提取技术与跨阶段多分支卷积CSP−OmniKernel(Cross Stage Partial−OmniKernel),通过优化特征金字塔网络的特征传递路径,有效减少小目标在特征融合阶段的信息丢失,增强了远距离、低分辨率目标的检测效果;同时,进一步调整了检测层结构,增加了高分辨率P2小目标检测层并删除了冗余的P5大目标检测层,使模型计算资源更集中于小目标检测任务;最后,引入了轻量化共享卷积检测头LSCD(Lightweight Shared Convolutional Detection),利用群组归一化替代批量归一化以提升小批量训练的稳定性,并通过共享卷积参数减少计算开销。自建井下数据集验证结果表明,改进模型在自建数据集上的平均精度均值mAP@0.5达到了83.6%,较原始YOLO11n提升了4.4%,参数量降低为2.474×106,检测速度达到了201.2 帧/s。与YOLOv5n、YOLOv8n、YOLOv10n等检测算法相比,改进模型在精度、轻量化上均表现更优,并且兼顾实时性,综合检测性能最佳。引入公共数据集DsLMF+验证改进模型的性能,模型的mAP@0.5达到96.2%,高于YOLOv5n、YOLOv8n、YOLO11n等算法,进一步验证了其良好的泛化能力,可为井下设备部署提供有效支持。

       

      Abstract: Aiming at the problems of low accuracy and high model complexity of small target detection algorithms for mine personnel safety helmets in complex mine environments, an improved detection algorithm based on YOLO11n is designed. First, a cross-stage partial-partial multi-scale feature aggregation (CSP-PMSFA) module is designed. This module uses 3×3, 5×5, and 7×7 convolution kernels of different sizes to extract features in parallel, and realizes effective fusion of multi-scale features through a partial selective connection mechanism, enhancing the ability to capture detailed features of safety helmets at multiple scales. Second, a small object enhance pyramid (SOEP) module is designed in the Neck layer. Combining the space-to-depth convolution (SPDConv) spatial feature extraction technology with cross-stage partial multi-branch convolution (CSP-OmniKernel), it optimizes the feature transmission path of the feature pyramid network, effectively reducing information loss of small targets during feature fusion, and enhancing the detection effect of long-distance and low-resolution targets. Meanwhile, the detection layer structure was further adjusted by adding a high-resolution P2 small-target detection layer and removing the redundant P5 large-target detection layer, concentrating the computing resources of model more on small-target detection tasks. Finally, a lightweight shared convolution detection head (LSCD) is introduced. Group normalization is used instead of batch normalization to improve the stability of small-batch training, and computing overhead is reduced by sharing convolution parameters. The results show that the average precision mean (mAP@0.5) of the improved model on the self-built dataset reaches 83.6%, which is 4.4% higher than that of the original YOLO11n, with the parameter count reduced to 2.474×106 and the detection speed reaching 201.2 frames per second (fps). Compared with detection algorithms such as YOLOv5n, YOLOv8n, and YOLOv10n, the improved model performs better in accuracy and lightweight design, while balancing real-time performance, achieving the best comprehensive detection performance. The performance of the improved model is verified using the public dataset DsLMF+, and the average precision mean (mAP@0.5) of the model reaches 96.2%, which is higher than algorithms such as YOLOv5n, YOLOv8n, and YOLO11n, further verifying its good generalization ability and providing effective support for the deployment of underground edge devices.

       

    /

    返回文章
    返回