Laser & Optoelectronics Progress, Volume. 58, Issue 6, 615002(2021)

Research on Helmet Wear Identification Based on Improved YOLOv3

Xu Kai and Deng Chao*
Author Affiliations
  • School of Physics & Electronic Information Engineering, Henan Polytechnic University, Jiaozuo, Henan454003,China
  • show less

    To address the problem of low detection accuracy and poor robustness of the traditional helmet wearing recognition algorithm, we propose a deep learning-based helmet wearing detection method. The method is based on the YOLOv3 detection algorithm, and its network structure and loss function are improved. Firstly, the poor detection of small targets by the original YOLOv3 algorithm is compensated by adding feature maps. Then, on the basis of adding feature maps, the K-means clustering algorithm is used to cluster the collected helmet datasets and select the appropriate a priori anchor frames. Finally, GIoU Loss is adopted as the boundary frame loss, and Focal Loss is added to the loss function to reduce the errors due to positive and negative sample imbalance. Experimental results show that, compared with the YOLOv3 detection algorithm, the improved algorithm improves the average accuracy by 3.47% and the accuracy of helmet identification by 4.23%, which is advanced and effective in helmet identification.

    Tools

    Get Citation

    Copy Citation Text

    Xu Kai, Deng Chao. Research on Helmet Wear Identification Based on Improved YOLOv3[J]. Laser & Optoelectronics Progress, 2021, 58(6): 615002

    Download Citation

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

    Category: Machine Vision

    Received: Jul. 24, 2020

    Accepted: --

    Published Online: Mar. 6, 2021

    The Author Email: Chao Deng (super@hpu.edu.cn)

    DOI:10.3788/LOP202158.0615002

    Topics