Abstract:Safety helmets are essential in various indoor and outdoor workplaces, such as metallurgical high-temperature operations and high-rise building construction, to avoid injuries and ensure safety in production. However, manual supervision is costly and prone to lack of enforcement and interference from other human factors. Moreover, small target object detection frequently lacks precision. Improving safety helmets based on the helmet detection algorithm can address these issues and is a promising approach. In this study, we proposed a modified version of the YOLOv5s network, a lightweight deep learning-based object identification network model. The proposed model extends the YOLOv5s network model and enhances its performance by recalculating the prediction frames, utilizing the IoU metric for clustering, and modifying the anchor frames with the K-means++ method. The global attention mechanism (GAM) and the convolutional block attention module (CBAM) were added to the YOLOv5s network to improve its backbone and neck networks. By minimizing information feature loss and enhancing the representation of global interactions, these attention processes enhance deep learning neural networks' capacity for feature extraction. Furthermore, the CBAM is integrated into the CSP module to improve target feature extraction while minimizing computation for model operation. In order to significantly increase the efficiency and precision of the prediction box regression, the proposed model additionally makes use of the most recent SIoU (SCYLLA-IoU LOSS) as the bounding box loss function. Based on the improved YOLOv5s model, knowledge distillation technology is leveraged to realize the light weight of the network model, thereby reducing the computational workload of the model and improving the detection speed to meet the needs of real-time monitoring. The experimental results demonstrate that the proposed model outperforms the original YOLOv5s network model in terms of accuracy (Precision), recall rate (Recall), and mean average precision (mAP). The proposed model may more effectively identify helmet use in low-light situations and at a variety of distances.

HDS-YOLOv5: An improved safety harness hook detection algorithm based on YOLOv5s

Toward Efficient Safety Helmet Detection Based on YoloV5 with Hierarchical Positive Sample Selection and Box Density Filtering

Improved YOLOv7 Electric Work Safety Belt Hook Suspension State Recognition Algorithm Based on Decoupled Head

SAFETY HELMET WEARING DETECTION BASED ON AN IMPROVED YOLOV3 SCHEME

HR-YOLO: A Multi-Branch Network Model for Helmet Detection Combined with High-Resolution Network and YOLOv5

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

Object Detection for Hazardous Material Vehicles Based on Improved YOLOv5 Algorithm

An improved YOLOv8 safety helmet wearing detection network

Site Safety Target Detection With Improved YOLOv5

YOLO-SDH: improved YOLOv5 using scaled decoupled head for object detection

Research on Safety Helmet Detection Algorithm Based on Improved YOLOv5s

DWCA-YOLOv5: an Improve Single Shot Detector for Safety Helmet Detection

YOLO-ESCA: A High-Performance Safety Helmet Standard Wearing Behavior Detection Model Based on Improved YOLOv5

YOLOV5s object detection based on Sim SPPF hybrid pooling

Real-time detection algorithm of helmet and reflective vest based on improved YOLOv5

YOLO-HR: Improved YOLOv5 for Object Detection in High-Resolution Optical Remote Sensing Images

SP-YOLO-Lite: A Lightweight Violation Detection Algorithm Based on SP Attention Mechanism

Workshop Safety Helmet Wearing Detection Model Based on SCM-YOLO

An improved method MSS-YOLOv5 for object detection with balancing speed-accuracy

Research on Real-Time Detection of Safety Harness Wearing of Workshop Personnel Based on YOLOv5 and OpenPose

Target Detection of Safety Protective Gear Using the Improved YOLOv5