高级检索

基于REIW-YOLOv10n的井下安全帽小目标检测算法

Small target detection algorithm for underground helmet based on REIW-YOLOv10n

  • 摘要: 针对井下工作面内光源、设备遮挡等多种因素引起的复杂工况环境所导致的作业人员安全帽小目标检测算法精度低等问题,在YOLOv10n模型的框架上,提出一种基于REIW-YOLOv10n的井下安全帽小目标图像检测算法。REIW-YOLOv10N模型包括Input,Backbone,Neck和Head 4个部分。为提高模型对多尺度特征的提取能力,设计RepNMSC结构,并在Backbone部分改进C2f结构,提升模型对多尺度的安全帽目标的特征提取能力;为保留Neck部分的小目标语义信息,Neck部分采用ERepGFPN结构,使用跨层连接的方式以相同的优先级处理高层语义信息和低层空间信息,实现融合小目标特征的目的;在Head部分添加P2小目标检测头并删去P5大目标检测头,在尽量保持轻量模型的前提下增加模型对井下安全帽小目标的检测性能;使用Inner-IoU和Wise-IoU v3的思想优化MPDIoU损失函数,Inner-Wise-MPDIoU使用缩放因子和梯度增益策略,加快模型收敛。利用CUMT-HelmeT数据集进行实验验证,REIW-YOLOv10n与YOLOv10n相比,mAP@0.5提升5.73%,达到88.24%。与YOLOv5s、YOLOv7-tiny、YOLOv8n和YOLOv9-tiny等5种YOLO系列算法和其他主流目标检测算法比较,REIW-YOLOv10n的精度和模型权重所占空间均优于其他对比算法,综合检测性能最佳。REIW-YOLOv10n在显著提高井下复杂环境检测安全帽小目标精度的前提下,兼顾了轻量化和实时性,方便模型部署在井下边缘设备。

     

    Abstract: To address the issue of low accuracy in detecting small targets, such as safety helmets, in complex underground working environments, caused by various factors like lighting conditions and equipment obstructions, a new image detection algorithm for small safety helmet targets based on the REIW-YOLOv10n model is proposed within the framework of the YOLOv10n model. The REIW-YOLOv10n model consists of four parts: Input, Backbone, Neck, and Head. To enhance the model’s ability to extract multi-scale features, the RepNMSC structure is designed, and the C2f structure in the Backbone section is improved, which boosts the model’s capability to extract features of multi-scale safety helmet targets. To preserve the small target semantic information in the Neck section, the ERepGFPN structure is adopted. This structure uses cross-layer connections to process high-level semantic information and low-level spatial information with the same priority, achieving the integration of small target features. Then, in the Head section, a P2 small target detection head is added, and the P5 large target detection head is removed. This improves the model’s detection performance for small safety helmet targets in underground environments while maintaining a lightweight model. Finally, the MPDIoU loss function is optimized using the concepts of Inner-IoU and Wise-IoU v3. Inner-Wise-MPDIoU employs scaling factors and gradient gain strategies to accelerate model convergence. Experiments using the CUMT-HelmeT dataset demonstrate that compared to YOLOv10n, REIW-YOLOv10n improves mAP@0.5 by 5.73%, reaching 88.24%. Compared with other mainstream YOLO series algorithms, such as YOLOv5s, YOLOv7-tiny, YOLOv8n, and YOLOv9-tiny, REIW-YOLOv10n outperforms them in terms of accuracy and model weight size, offering the best overall detection performance. REIW-YOLOv10n significantly improves the accuracy of detecting small safety helmet targets in complex underground environments while balancing lightweight design and real-time processing, making it convenient for deployment on underground edge devices.

     

/

返回文章
返回