Safety Helmet Wearing Detection Method Fused with Self-Attention Mechanism

doi:10.3778/j.issn.1002-8331.2103-0372

Abstract

Abstract: Wearing a safety helmet is one of the effective methods to prevent head injury of workers on the construction site. However, the existing safety helmet detection algorithms mostly have disadvantages such as the difficulty of detecting overlapping targets and the high rate of missed detection of small targets. For this reason, an improved Faster R-CNN target detection algorithm by fusing the self-attention mechanism is proposed for helmet detection. First, the self-attention layer is used to capture global information on multiple scales to obtain richer high-level semantic features and introduce a larger range of receptive fields into the model, and then through the anchor box by-selection in the training of the regional proposal network（RPN）, the enhanced method allows more training of small target information, and strengthens the network’s ability to express small-scale targets. The experimental results show that the improved algorithm has a 6.4 percentage points increase in the mAP value of the helmet wearing detection compared to the traditional Faster R-CNN, and it has a better detection effect for helmets in different scenarios and different scales.

Key words: small target detection, Faster R-CNN algorithm, self-attention mechanism, helmet wearing recognition

摘要： 佩戴安全帽是防止施工现场工作人员头部损伤的有效方法之一，然而现有安全帽检测算法多存在重叠目标检测难度大、小目标漏检率高等缺点。为此，提出了一种通过融合自注意力机制来改进Faster R-CNN的目标检测算法，用于安全帽检测。通过自注意力层来捕获多个尺度上的全局信息，得到更丰富的高层语义特征并将更大的感受野范围引入模型，在区域建议网络（RPN）的训练中通过锚框补选增强的方法让小目标信息得到更多的训练，强化了网络对于小尺度目标的表达能力。实验结果表明：改进后的算法在安全帽佩戴检测上的mAP值较传统Faster R-CNN提高了6.4个百分点，对于不同场景不同尺度的安全帽有着较好的检测效果。

关键词: 小目标检测, Faster R-CNN算法, 自注意力机制, 安全帽佩戴识别

SUN Guodong, LI Chao, ZHANG Hang. Safety Helmet Wearing Detection Method Fused with Self-Attention Mechanism[J]. Computer Engineering and Applications, 2022, 58(20): 300-304.

孙国栋, 李超, 张航. 融合自注意力机制的安全帽佩戴检测方法[J]. 计算机工程与应用, 2022, 58(20): 300-304.

References

[1] NAJIBI M，SAMANGOUEI P，CHELLAPPA R，et al.SSH：single stage headless face detectorn[C]//Proceedings of the IEEE International Conference on Computer Vision，2017：4875-4884.
[2] ZHANG L，LIN L，LIANG X，et al.Is Faster R-CNN doing well for pedestrian detection[C]//European Conference on Computer Vision，2016：443-457.
[3] CHEN Q，TANG S，YANG Q，et al.Cooper：cooperative perception for connected autonomous vehicles based on 3D point clouds[C]//International Conference on Distri-
buted Computing Systems，2019：514-524.
[4] WANG Z，FU H，WANG L，et al.SCNet：subdivision coding network for object detection based on 3D point cloud[J].IEEE Access，2019，7（99）：449-462.
[5] RAGHUNANDAN A，MOHANA，PAKALA R，et al.Object detection algorithms for video surveillance applications[C]//International Conference on Communication and Signal Processing，2018：563-568.
[6] 刘云波，黄华.施工现场安全帽佩戴情况监控技术研究[J].电子科技，2015，28（4）：69-72.
LIU Yunbo，HUANG Hua.Research on monitoring techno-
logy of wearing safety helmet on construction site[J].Electronic Science and Technology，2015，28（4）：69-72.
[7] 刘晓慧，叶西宁.肤色检测和Hu矩在安全帽识别中的应用[J].华东理工大学学报（自然科学版），2014，40（3）：365-370.
LIU Xiaohui，YE Xining.Application of skin color detection and Hu moment in helmet recognition[J].Journal of East China University of Science and Technology（Natural Science Edition），2014，40（3）：365-370.
[8] KISHOR S，SHRESTHA P P，DINESH B，et al.Hard-hat detection for construction safety visualization[J].Journal of Construction Engineering，2015：1-8.
[9] GIRSHICK R，DONAHUE J，DARRELL T，et al.Region-based convolutional networks for accurate object detection and segmentation[J].IEEE Transactions on Pattern Analysis and Machine Intelligence，2015：142-158.
[10] GIRSHICK R.Fast R-CNN[C]//Proceedings of the IEEE International Conference on Computer Vision，2015：1440-1448.
[11] REN S，HE K，GIRSHICK R，et al.Faster R-CNN：towards real-time object detection with region proposal networks[J].IEEE Transactions on Pattern Analysis & Machine Intelligence，2015，39（6）：1137-1149.
[12] REDMON J，FARHADI A.YOLO9000：better，faster，stronger[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2017：7263-7271.
[13] REDMON J，FARHADI A.YOLOv3：an incremental improvement[J].arXiv：1804.02767，2018.
[14] LIU W，ANGUELOV D，ERHAN D，et al.SSD：single shot multibox detector[C]//European Conference on Computer Vision.Cham：Springer，2016：21-37.
[15] 王兵，李文璟，唐欢.改进YOLO v3算法及其在安全帽检测中的应用[J].计算机工程与应用，2020，56（9）：33-40.
WANG Bing，LI Wenjing，TANG Huan.Improved YOLO v3 algorithm and its application in helmet detection[J].Computer Engineering and Applications，2020，56（9）：33-40.
[16] 张博，宋元斌，熊若鑫，等.融合人体关节点的安全帽佩戴检测[J].中国安全科学学报，2020，30（2）：181-186.
ZHANG Bo，SONG Yuanbin，XIONG Ruoxin，et al.Safety helmet wearing detection based on human joint points[J].Chinese Journal of Safety Science，2020，30（2）：181-186.
[17] LIN T Y，DOLLáR P，GIRSHICK R，et al.Feature pyramid networks for object detection[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2017：2117-2125.
[18] LIU X，LI K，LI K.Attentive semantic and perceptual faces completion using self-attention generative adversarial networks[J].Neural Processing Letters，2019：1-19.