Reflective clothing detection based on YOLOv5s fused with SE attention mechanism

This paper proposes a YOLOv5s deep learning algorithm incorporating the SE attention mechanism to address the issue of workers failing to wear reflective clothing on duty, which has resulted in casualties from time to time.The YOLOv5s model is used to train the reflective clothing dataset obtained by collecting images of construction sites, and the SE attention mechanism module is added to the network structure of YOLOv5 for improved performance.The reflective clothing detection algorithm is then deployed to the web for real-time detection.The experimental results show that the algorithm achieves mAP value of 96.70% with the SE attention module, resulting in an accuracy improvement of 1.09% compared to the model without the SE module.This enables practical applications to reduce the incidence of construction accidents at construction sites.