Personal Protective Equipment Detection for Construction Workers: A Novel Dataset and Enhanced YOLOv5 Approach

Abstract

Current research on personal protective equipment (PPE) detection has mainly focused on hard hats, overlooking the detection of reflective clothing. Therefore, this study aims to address this research gap comprehensively. We achieve this by creating a novel dataset using semi-automatic labeling techniques and enhancing the YOLOv5 model. The dataset consists of four categories, assessing the presence of hard hats and reflective clothing. Additionally, we introduce an attention mechanism and an improved loss function to tackle challenges related to detecting reflective clothing and overlapping detection frames. Through extensive multi-model comparison experiments, our improved model, AL-YOLOv5, outperforms the baseline model with remarkable advancements of 0.9 AP in the data-limited category and 0.4 mAP overall. Notably, our improved model shows substantial progress in detecting reflective clothing, significantly reducing false detections, and improving overlapping bounding frames. In conclusion, this study contributes to PPE detection accuracy through a novel dataset and an improved model.

Keywords

• Attention mechanisms
• new dataset
• personal protective equipment (PPE)
• reflective clothing
• SEIoU