Skip to content
Conference

Lightweight safety helmet detection algorithm for smart construction sites based on improved yolov8

Jul 2026 · International Conference on Machine Vision and Applications · Vol 14270, pp. 142700W - 142700W-6 · 0 citations · 4 references
Engineering

TL;DR

A lightweight detection algorithm named GCW-YOLOv8, which improves detection accuracy by 1.4% while reducing parameter count by 34.4%, achieving a superior balance between accuracy and efficiency for smart construction site applications.

Abstract

Safety helmet wearing detection is a crucial component of safety management in construction sites. Traditional detection methods based on manual monitoring are inefficient, while existing deep learning models often suffer from high computational costs and poor detection performance for small or occluded targets. To address these issues, this paper proposes a lightweight detection algorithm named GCW-YOLOv8. Firstly, the Ghost Module is introduced into the backbone network to replace the conventional convolution layers, significantly reducing the number of parameters and floating-point operations (FLOPs) while maintaining feature extraction capability. Secondly, the Coordinate Attention (CA) mechanism is embedded into the neck network to enhance the model's sensitivity to spatial location and channel information, thereby improving the detection accuracy of small targets. Finally, the Wise-IoU (WIoU) loss function is adopted to replace the original CIoU loss, utilizing a dynamic non-monotonic focusing mechanism to optimize the gradient assignment for low-quality samples. Experimental results on the safety helmet dataset show that the proposed algorithm achieves a mean Average Precision (mAP@0.5) of 94.5%, while the inference speed reaches 108 FPS. Compared with the baseline YOLOv8n, our method improves detection accuracy by 1.4% while reducing parameter count by 34.4%, achieving a superior balance between accuracy and efficiency for smart construction site applications.

View source

Similar papers

Aug 2026

XLiteYOLOv12: a lightweight framework for real-time hazardous object detection

The YOLOv12 network is adopted as the baseline model and the ADown module is introduced to improve downsampling efficiency while maintaining lightweight performance, and the BN-CGLU is incorporated into the A2C2f module to enhance the model’s nonlinear representation capability.

Aobo Yue, Puchun Chen, Yan Yang · 0 citations
Conference Jul 2026

YOLOv8 vehicle detection method based on lightweight coordinate attention

To address the challenges encountered in vehicle detection within complex traffic scenarios, such as diverse features, strong background interference, and deployment constraints on edge devices, a lightweight vehicle detection method based on YOLOv8 improved with the Coordinate Attention mechanism is proposed. In this method, the Coordinate Attention module is introduced into the YOLOv8s backbone network to reconstruct the C2f structure. By embedding positional information, the model enhances its capability to extract key features, suppressing complex background interference while maintaining its lightweight characteristics. Experiments conducted on a dataset comprising 17,428 images of 17 vehicle classes demonstrate that the improved model achieves a precision of 90.4% and an mAP@0.5 of 91.5%. For vehicle types with regular structures and large volumes, such as four-wheel small trucks and six-wheel medium trucks, the detection accuracy approaches 99%, and the detection confidence remains stable above 0.8 under complex lighting and background interference conditions. This method effectively controls model complexity while maintaining high detection accuracy, thereby satisfying the application requirements for real-time vehicle detection on edge devices such as intelligent driving recorders.

Zhihan Wang · 0 citations
Conference Jul 2026

Research on lightweight detection of maritime ship targets based on iGF-YOLOv11

To address the difficulty of effectively detecting maritime ship targets on mobile devices with limited computing resources under interference from illumination variations and wave disturbances, this paper proposes a lightweight ship detection model, termed iGF-YOLOv11, based on YOLOv11. Firstly, the GhostConv module is used to replace the standard convolution in the backbone network to reduce the model complexity and computational load; secondly, the iEMA attention module that integrates the advantages of iRMB and EMA mechanisms is introduced in the neck network to enhance the model's ability to extract key features of ships; moreover, a P2 detection head is added to improve the perception performance of small ship targets. Experimental results show that the improved model maintains high efficiency while significantly improving the detection accuracy. The proposed method achieves a balance between detection accuracy and lightweighting, providing a feasible solution for the deployment and operation of detection models on mobile devices.

Kairen Chen, Shengwei Xing, Rui Qi · 0 citations
Jul 2026

A lightweight detection method for defects of transmission line insulators and accessories based on improved YOLOv11n

Insulators and their accessories are core components ensuring the safe operation of transmission lines. Defect detection for them is confronted with challenges such as small targets, complex backgrounds and difficulty in fitting the irregular boundaries of defects, while traditional lightweight algorithms suffer from insufficient detection performance. To tackle these challenges, this work presents a lightweight detection algorithm built upon the improved YOLOv11n: the DCNv3 (Deformable Convolution v3) is embedded into the backbone network to replace the traditional convolutional layers, enhancing the ability to extract features of irregular defects; the Inner-WIoU(Inner-Weighted Intersection over Union) loss function is employed to refine the bounding box regression. All experiments are conducted on a self-built dataset with 7,972 images and 7 defect categories. The proposed method achieves a mAP@50 of 86.4%, 3.2 percentage points higher than the baseline YOLOv11n, and outperforms mainstream lightweight detectors and small-object-oriented detectors. Meanwhile, the algorithm retains the lightweight advantage and meets the needs of on-site real-time inspection. The model maintains a lightweight structure with only 2.6 M parameters and achieves an inference speed of 38 FPS on NVIDIA RTX 4080 Super, meeting real-time inspection requirements. Meanwhile, the algorithm retains the lightweight advantage and can meet the requirements of on-site real-time inspection. Since validation is limited to a self-constructed dataset, cross-scene testing and public benchmark evaluation will be conducted in future work to further verify the model’s robustness and generalization in real transmission-line inspection scenarios.

Yu-Jie Sheng, Junjie Sang, Liutao Lu · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.