Chinese Journal of Liquid Crystals and Displays, Volume. 38, Issue 7, 964(2023)

Improved lightweight helmet wear detection algorithm

Xue-chun LIU1,2, Da-ming LIU1,2、*, and Ruo-chen LIU1,2
Author Affiliations
  • 1School of Electronics and Electrical Engineering,Ningxia University,Yinchuan 750000,China
  • 2Key Laboratory of Intelligent Sensing of Desert Information,Ningxia University,Yinchuan 750000,China
  • show less

    For the existing helmet wearing detection algorithm for dense targets and small targets having the phenomenon of missed detection, many parameters, large computation, and not suitable for deployment in the embedded device side and other problems, this paper proposes an improved YOLOv5 helmet wearing detection algorithm YOLOv5-Q. Firstly, on the original network 80×80 feature map, the 2 times up-sampling operation is made to form a 160×160 feature map, and the new feature map fuses the three-layer feature information of the original model to form a four-scale detection, which improves the detection accuracy of dense targets and small targets. Secondly, the feature extraction is achieved by replacing the original YOLOv5 backbone network with a lightweight GhostNet, which reduces the parameters of the network and can be ported to the embedded devices for target detection. Finally, the attention mechanism CA is added to boost the weight of important information in the feature map and suppress the weight of non-relevant information, thus improving the accuracy of the model. The experimental results show that the model size of YOLOv5-Q is 26.47 MB, the number of parameters is 12 696 640, and the accuracy is 0.937. Compared with YOLOv5, the YOLOv5-Q algorithm reduces the number of parameters by 39.12%, the model size by 37.2%, but the accuracy by only 1.2%. The YOLOv5-Q algorithm increases the detection accuracy of small targets in dense environments and meets the requirements for deployment on the embedded side.

    Tools

    Get Citation

    Copy Citation Text

    Xue-chun LIU, Da-ming LIU, Ruo-chen LIU. Improved lightweight helmet wear detection algorithm[J]. Chinese Journal of Liquid Crystals and Displays, 2023, 38(7): 964

    Download Citation

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

    Category: Research Articles

    Received: Aug. 13, 2022

    Accepted: --

    Published Online: Jul. 31, 2023

    The Author Email: Da-ming LIU (nxldm@qq.com)

    DOI:10.37188/CJLCD.2022-0268

    Topics