基于改进YOLOv8s的井下安全帽检测算法

    Underground helmet detection algorithm based on improved YOLOv8s

    • 摘要: 煤矿井下作业过程中,安全帽是防护措施中最直接最有效的措施之一,是矿工生命安全的重要保障。安全帽属于小目标检测,而井下作业环境复杂,粉尘和光线噪声对摄像头检测有极大的影响。针对上述问题,提出了一种基于改进YOLOv8s的井下安全帽的检测算法,称为PBSS-YOLOv8。PBSS-YOLOv8模型首先添加了小目标检测层P2,用于提升小目标的检测性能,并在此基础上引入BiFPN网络框架,使信息在网络中可以双向传输,进一步提高对安全帽的捕获性能,并删除了原始P5检测层,极大地降低模型的参数量、计算量;之后加入SPD-Conv卷积模块,使用非步长的卷积层,降低小目标的冗余信息被过滤、细粒度信息丢失的情况;最后加入轻量化注意力机制SGE,降低环境和噪声对目标特征提取的影响。试验结果表明:改进后的PBSS-YOLOv8较YOLOv8s,Helmet AP提高了1.1%,No-Helmet AP提高了3.5%,mAP提高了2.8%,参数量下降了2 M。使用煤矿井下监控系统对改进后的模型进行验证,其有效地改善了小目标漏检和误检的问题,为矿下人员的安全作业提供了保障。

       

      Abstract: In the process of coal mine underground operation, the safety helmet is the most direct and effective protective measure, and it is an important measure to ensure the safety of miners. The safety helmet belongs to small target detection, and the underground operation environment is complex, and dust and light noise have a great impact on camera detection. In order to solve the above problems, this study proposes a detection algorithm for underground safety helmets based on improved YOLOv8s, which is called PBSS-YOLOv8. The PBSS-YOLOv8 model first adds the small target detection layer P2 to improve the detection performance of small targets, and on this basis, the BiFPN network framework is introduced to make the information in the network can be transmitted in both directions, which further improves the capture performance of the helmet, and the original P5 detection layer is deleted, which greatly reduces the number of parameters and calculations of the model. Then, the SPD-Conv convolution module is added, and the non-step convolutional layer is used to reduce the situations that the redundant information of small targets is filtered and fine-grained information is lost. Finally, the lightweight attention mechanism SGE is added to reduce the influence of environment and noise on target feature extraction. Experimental results show that compared with YOLOv8s, the improved PBSS-YOLOv8 increases the Helmet AP by 1.1%, the No-Helmet AP by 3.5%, the mAP by 2.8%, and the parameter amount decreases by 2 M. The experimental results confirm that the improved model can effectively improve the problem of missed detection and false detection of small targets, and provide a guarantee for the safe operation of underground personnel.

       

    /

    返回文章
    返回