An Improved Lightweight Safety Helmet Detection Algorithm for YOLOv8  

作　　者：Lieping Zhang Hao Ma Jiancheng Huang Cui Zhang Xiaolin Gao 

机构地区：[1]Key Laboratory of AdvancedManufacturing and Automation Technology,EducationDepartment of Guangxi Zhuang Autonomous Region,Guilin University of Technology,Guilin,541004,China [2]Guangxi Key Laboratory of Special Engineering Equipment and Control,Guilin University of Aerospace Technology,Guilin,541004,China [3]Guangxi Tianli Construction Engineering Co.,Ltd.,Guilin,541001,China [4]School of Information Engineering,Nanning College of Technology,Guilin,541004,China

出　　处：《Computers, Materials & Continua》2025年第5期2245-2265,共21页计算机、材料和连续体(英文)

基　　金：funded by National Natural Science Foundation of China(61741303);the Foundation Project of Guangxi Key Laboratory of Spatial Information andMapping(No.21-238-21-16).

摘　　要：Detecting individuals wearing safety helmets in complex environments faces several challenges.These factors include limited detection accuracy and frequent missed or false detections.Additionally,existing algorithms often have excessive parameter counts,complex network structures,and high computational demands.These challenges make it difficult to deploy such models efficiently on resource-constrained devices like embedded systems.Aiming at this problem,this research proposes an optimized and lightweight solution called FGP-YOLOv8,an improved version of YOLOv8n.The YOLOv8 backbone network is replaced with the FasterNet model to reduce parameters and computational demands while local convolution layers are added.This modification minimizes computational costs with only a minor impact on accuracy.A new GSTA(GSConv-Triplet Attention)module is introduced to enhance feature fusion and reduce computational complexity.This is achieved using attention weights generated from dimensional interactions within the feature map.Additionally,the ParNet-C2f module replaces the original C2f(CSP Bottleneck with 2 Convolutions)module,improving feature extraction for safety helmets of various shapes and sizes.The CIoU(Complete-IoU)is replaced with the WIoU(Wise-IoU)to boost performance further,enhancing detection accuracy and generalization capabilities.Experimental results validate the improvements.The proposedmodel reduces the parameter count by 19.9% and the computational load by 18.5%.At the same time,mAP(mean average precision)increases by 2.3%,and precision improves by 1.2%.These results demonstrate the model’s robust performance in detecting safety helmets across diverse environments.

关 键 词：YOLO safety helmet detection complex environments LIGHTWEIGHT WIoU loss function 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术]