Target Detection of Safety Protective Gear Using the Improved YOLOv5

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Liu, Hao, Qin, Xue
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909377538031616
author Liu, Hao
Qin, Xue
author_facet Liu, Hao
Qin, Xue
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2408_05964
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Target Detection of Safety Protective Gear Using the Improved YOLOv5
Liu, Hao
Qin, Xue
Computer Vision and Pattern Recognition
Machine Learning
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.
title Target Detection of Safety Protective Gear Using the Improved YOLOv5
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2408.05964