Better YOLO with Attention-Augmented Network and Enhanced Generalization Performance for Safety Helmet Detection

arXiv:2405.02591v1 Announce Type: new
Abstract: Safety helmets play a crucial role in protecting workers from head injuries in construction sites, where potential hazards are prevalent. However, currently, there is no approach that can simultaneously achieve both model accuracy and performance in complex environments. In this study, we utilized a Yolo-based model for safety helmet detection, achieved a 2% improvement in mAP (mean Average Precision) performance while reducing parameters and Flops count by over 25%. YOLO(You Only Look Once) is a …

abstract accuracy arxiv attention construction cs.cv detection environments hazards head however model accuracy network performance role safety study type workers yolo