Open Access iconOpen Access

ARTICLE

crossmark

Safety Helmet Wearing Detection in Aerial Images Using Improved YOLOv4

Wei Chen1, Mi Liu1,*, Xuhong Zhou2, Jiandong Pan3, Haozhi Tan4

1 College of Civil Engineering, Changsha University of Science and Technology, Changsha, 410114, China
2 College of Civil Engineering, Chongqing University, Chongqing, 730000, China
3 College of Mechanical and Electrical Engineering, Hunan Agricultural University, Changsha, 410128, China
4 Department of Civil and Environmental Engineering, University of Auckland, Auckland, 1010, New Zealand

* Corresponding Author: Mi Liu. Email: email

Computers, Materials & Continua 2022, 72(2), 3159-3174. https://doi.org/10.32604/cmc.2022.026664

Abstract

In construction, it is important to check whether workers wear safety helmets in real time. We proposed using an unmanned aerial vehicle (UAV) to monitor construction workers in real time. As the small target of aerial photography poses challenges to safety-helmet-wearing detection, we proposed an improved YOLOv4 model to detect the helmet-wearing condition in aerial photography: (1) By increasing the dimension of the effective feature layer of the backbone network, the model's receptive field is reduced, and the utilization rate of fine-grained features is improved. (2) By introducing the cross stage partial (CSP) structure into path aggregation network (PANet), the calculation amount of the model is reduced, and the aggregation efficiency of effective features at different scales is improved. (3) The complexity of the YOLOv4 model is reduced by introducing group convolution and the pruning PANet multi-scale detection mode for de-redundancy. Experimental results show that the improved YOLOv4 model achieved the highest performance in the UAV helmet detection task, that the mean average precision (mAP) increased from 83.67% of the original YOLOv4 model to 91.03%, and that the parameter amount of the model is reduced by 24.7%. The results prove that the improved YOLOv4 model can effectively respond to the requirements of real-time detection of helmet wearing by UAV aerial photography.

Keywords


Cite This Article

W. Chen, M. Liu, X. Zhou, J. Pan and H. Tan, "Safety helmet wearing detection in aerial images using improved yolov4," Computers, Materials & Continua, vol. 72, no.2, pp. 3159–3174, 2022.



cc This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 1474

    View

  • 937

    Download

  • 0

    Like

Share Link