An improved personal protective equipment detection method based on YOLOv4
Rengjie Qiao,Chengtao Cai,Haiyang Meng,Kejun Wu,Feng Wang,Jie Zhao
DOI: https://doi.org/10.1007/s11042-024-18772-1
IF: 2.577
2024-03-15
Multimedia Tools and Applications
Abstract:Personal protective equipment (PPE) detection plays a crucial role in ensuring safety in various settings such as factories, hospitals, and disease prevention measures. However, manually checking individuals for proper PPE usage in public can be a challenging task. This paper focuses on the detection of face mask usage and aims to develop a robust system that can identify individuals who are not wearing masks or are not wearing them correctly. Here, we present an enhanced face mask detection method based on YOLOv4. Currently, there is a shortage of a comprehensive and diverse dataset that can be used to accurately evaluate the correct use of masks, mainly due to limited samples of incorrect mask wearing. To address this issue, we propose a pipeline to generate a simulated face mask dataset derived from the original dataset. This approach allows us to enhance the performance of the face mask detection model without requiring additional data samples. Additionally, we introduce a modified face mask detection model called MaskYOLO, which includes improvements in the original YOLOv4 network structure. In the feature extraction network, a global context block is incorporated between the backbone and neck of YOLOv4 to obtain a more comprehensive understanding of the scene. Furthermore, the prediction network incorporates an improved structure to achieve a more efficient network. The effectiveness and accuracy of the proposed method are demonstrated through statistical analyses of the experimental results. Our method outperforms the YOLOv4 baseline by 3.1% in mean Average Precision (mAP).
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering