Laser & Optoelectronics Progress, Volume. 60, Issue 12, 1210011(2023)

Improved YOLOv4 Helmet Detection Algorithm Under Complex Scenarios

Guobo Xie1, Jingjing Tang1, Zhiyi Lin1、*, Xiaofeng Zheng1, and Ming Fang2
Author Affiliations
  • 1School of Computer Science and Technology, Guangdong University of Technology, Guangzhou 510006, Guangdong, China
  • 2Transmission Branch of Yunnan Power Grid Co., Ltd., Kunming 650011, Yunnan, China
  • show less

    An improved helmet detection algorithm for YOLOv4 (SMD-YOLOv4) is proposed to effectively detect whether construction workers are wearing helmets in complex scenes and reduce safety hazards. First, the SE-Net attention module is used to improve the ability of the model backbone network to extract effective features. Next, a dense atrous space pyramid pooling (DenseASPP) is used instead of spatial pyramid pooling (SPP) in the network to reduce information loss and optimize the extraction of global contextual information. Finally, the scale of feature fusion is increased in the PANet part and deep separable convolution is introduced to obtain detailed information about small targets in complex contexts without slowing down the network inference speed. The experimental results show that the mean average precision (mAP) of SMD-YOLOv4 algorithm reaches 97.34% on the self-built experimental dataset, which is 26.41 percentage points, 6.44 percentage points, 3.25 percentage points, 1.49 percentage points, and 3.19 percentage points higher than that of the current representative Faster R-CNN, SSD, YOLOv5, YOLOx, and original YOLOv4 algorithms, respectively, and can meet the real-time detection requirements.

    Tools

    Get Citation

    Copy Citation Text

    Guobo Xie, Jingjing Tang, Zhiyi Lin, Xiaofeng Zheng, Ming Fang. Improved YOLOv4 Helmet Detection Algorithm Under Complex Scenarios[J]. Laser & Optoelectronics Progress, 2023, 60(12): 1210011

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Apr. 22, 2022

    Accepted: Jun. 16, 2022

    Published Online: May. 23, 2023

    The Author Email: Lin Zhiyi (lzy291@gdut.edu.cn)

    DOI:10.3788/LOP221388

    Topics