Open Access

ARTICLE

Safety Helmet Wearing Detection in Aerial Images Using Improved YOLOv4

by Wei Chen¹, Mi Liu^1,*, Xuhong Zhou², Jiandong Pan³, Haozhi Tan⁴

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:

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

Received 31 December 2021; Accepted 09 February 2022; Issue published 29 March 2022

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

---