基于YOLOX的安全帽与口罩检测算法

SAFETYHELMETANDMASKDETECTIONALGORITHMBASEDONYOLOX

  • 摘要: 针对当前基于深度学习的安全帽与口罩检测算法存在精确度不高,模型庞大等不足,提出基于YOLOX-S的安全帽与口罩的检测方法。该方法在YOLOX-S算法的基础上结合全局上下文模块(GCblock)与特征增强模块(RFB),接着将浅层有效特征层与注意力模块融入加强特征提取网络中,最后更换Mish激活函数。以上改进点的加入使得模型的信息提取能力得到增强,经实验验证得到,改进后的网络模型的mAP值达到了87.18%,比原始的YOLOX-S网络模型的mAP值提升了1.07百分点。

     

    Abstract: Deep learning- based algorithms for detecting safety helmets and masks often suffer from low accuracy and large model sizes. To address these limitations, this paper proposes a method for detecting safety helmets and masks based on YOLOX- S. This method combined a global contextual block (GCblock) and a feature enhancement module (RFB) based on the YOLOX- S algorithm. It incorporated shallow effective feature layers and attention modules into the enhanced feature extraction network, and replaced the Mish activation function. These improvements enhanced the model's information extraction ability. Experimental results show that the modified network model achieves the mAP value of 87.18%, which is a 1.07 percentage points improvement over the original YOLOX- S network model.

     

/

返回文章
返回