Abstract:Workplace safety accidents are a pervasive issue worldwide. According to the National Work Safety Supervision Administration, a striking 67.95 % of construction accidents occur due to workers not wearing helmets. Existing helmet-wearing detection algorithms, however, tend to underperform in real-world scenarios where challenges such as smaller helmet areas in images, complex backgrounds, and object occlusions are present. Additionally, these models have a considerable amount of parameters, which impedes their practical deployment. This study proposes a novel, lightweight helmet detection algorithm, YOLO-PL, based on YOLOv4, to address these challenges. Initially, we designed the YOLO-P algorithms. YOLO-P algorithms optimize the network structure by refining its ability to detect small objects and improving the anchor assignment in the detection head. We design the Enhanced PAN (E-PAN) structure to merge the higher-layer, low-noise information with the lower-layer information based on the Path Aggregation Network (PAN). The YOLO-P algorithm improves detection accuracy by using the E-PAN structure. Subsequently, while preserving the performance of the YOLO-P algorithm, we enhanced its design for lightness. We proposed the Dilated Convolution Cross Stage Partial with X res units (DCSPX) module based on the Cross Stage Partial (CSP) structure, replacing the Spatial Pyramid Pooling (SPP) module with it. Additionally, we designed a Lightweight VoVNet (L-VoVN) structure based on the architecture of VoVNet, introduced a lightweight Max-Pooling (MP) down-sampling method, and fine-tuned the Swish activation function, which led to the final YOLO-PL algorithm. YOLO-PL significantly reduces the parameters in YOLO-P, thus achieving state-of-the-art performance that surpasses current object detectors like YOLOv5 and v7 in safety helmet detection. Moreover, our model exhibits substantial improvements in robustness and deployability, demonstrating considerable potential for practical implementations in industry.

PLOD-YOLO: Premium Lightweight Object Detection for Autonomous Following Robot

Lightweight object detection algorithm for robots with improved YOLOv5

Lightweight Low-Altitude UAV Object Detection Based on Improved YOLOv5s

A YOLO-NL object detector for real-time detection

YOLO_SRv2: An evolved version of YOLO_SR

An improved YOLOv8 algorithm for small object detection in autonomous driving

DAN-YOLO: A Lightweight and Accurate Object Detector Using Dilated Aggregation Network for Autonomous Driving

LAM-YOLO: Drones-based Small Object Detection on Lighting-Occlusion Attention Mechanism YOLO

High-precision real-time autonomous driving target detection based on YOLOv8

DCW-YOLO: Road Object Detection Algorithms for Autonomous Driving

YOLO-DRS: A Bioinspired Object Detection Algorithm for Remote Sensing Images Incorporating a Multi-Scale Efficient Lightweight Attention Mechanism

BiGA-YOLO: A Lightweight Object Detection Network Based on YOLOv5 for Autonomous Driving

MobileYOLO: Real-Time Object Detection Algorithm in Autonomous Driving Scenarios

YOLOD: A Target Detection Method for UAV Aerial Imagery

YOLO-PL: Helmet wearing detection algorithm based on improved YOLOv4

A Lightweight Object Detection Algorithm for Remote Sensing Images Based on Attention Mechanism and YOLOv5s

Lightweight aerial image object detection algorithm based on improved YOLOv5s

YOLOv10: Real-Time End-to-End Object Detection

Lightweight unmanned aerial vehicle object detection algorithm based on improved YOLOv8

YOLOv4-5D: An Effective and Efficient Object Detector for Autonomous Driving

RS-YOLO: A YOLO-Based Method for Small Object Detection in Remote Sensing