Target Detection of Safety Protective Gear Using the Improved YOLOv5
In high-risk railway construction, personal protective equipment monitoring is critical but challenging due to small and frequently obstructed targets. We propose YOLO-EA, an innovative model that enhances safety measure detection by integrating ECA into its backbone's convolutional layers, improving discernment of minuscule objects like hardhats. YOLO-EA further refines target recognition under occlusion by replacing GIoU with EIoU loss. YOLO-EA's effectiveness was empirically substantiated using a dataset derived from real-world railway construction site surveillance footage. It outperforms YOLOv5, achieving 98.9% precision and 94.7% recall, up 2.5% and 0.5% respectively, while maintaining real-time performance at 70.774 fps. This highly efficient and precise YOLO-EA holds great promise for practical application in intricate construction scenarios, enforcing stringent safety compliance during complex railway construction projects.
Authors
- Hao Liu (ORCID: https://orcid.org/0009-0003-4676-1796)
- Xue Qin (ORCID: https://orcid.org/0000-0002-4513-3515)
Institutions
- Guizhou University (CN)
Publication Details
- Published
- 2024-12-20
- DOI
- https://doi.org/10.1109/cait64506.2024.10962947
- Citations
- 4
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 1.40