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

TL;DR

This paper introduces an improved YOLO-based safety helmet detection model that combines GhostNetv2, attention modules, and a specialized optimizer to enhance accuracy, efficiency, and adaptability in complex environments.

Contribution

It presents a novel lightweight network with attention mechanisms and a new optimizer, achieving higher accuracy and better generalization for helmet detection.

Findings

2% improvement in mAP over baseline

Reduced parameters and Flops by over 25%

Enhanced robustness and speed in complex scenes

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 widely used, high-performance, lightweight model architecture that is well suited for complex environments. We presents a novel approach by incorporating a lightweight feature extraction network backbone based on GhostNetv2, integrating attention modules such as Spatial Channel-wise Attention Net(SCNet) and Coordination Attention Net(CANet), and adopting the Gradient Norm Aware optimizer (GAM) for improved generalization ability. In safety-critical environments, the accurate detection and speed of safety helmets plays a pivotal role in preventing occupational hazards and ensuring compliance with safety protocols. This work addresses the pressing need for robust and efficient helmet detection methods, offering a comprehensive framework that not only enhances accuracy but also improves the adaptability of detection models to real-world conditions. Our experimental results underscore the synergistic effects of GhostNetv2, attention modules, and the GAM optimizer, presenting a compelling solution for safety helmet detection that achieves superior performance in terms of accuracy, generalization, and efficiency.