计算机与现代化Issue(3)：95-101,7.DOI:10.3969/j.issn.1006-2475.2026.03.013
基于ESH-YOLO的小目标安全帽检测算法
Small Target Helmet Detection Algorithm Based on ESH-YOLO
摘要
Abstract
Aiming at the problems of low detection accuracy,false detection and missed detection of existing safety helmet wear-ing detection algorithms for small targets in complex scenes,this paper proposes an improved safety helmet wearing detection al-gorithm ESH-YOLO based on YOLOv8n.Firstly,the Star_CAA module is obtained by combining the Star Blocks network with the CAA attention mechanism,and the module is used to optimize the C2f structure in the original network backbone to enhance the recognition performance of the model for safety helmets in complex scene images,giving more attention weight to small target safety helmets and inhibiting the attention to background and worker regions.Secondly,ELA_HSPAN is used as the neck struc-ture.Through feature selection and fusion of feature maps with different resolutions,the feature information of each level is better relayed,and the detection and localization ability of objects with different scales,especially small objects,is improved.Finally,the PIoUv2 loss function is introduced to accelerate the convergence speed of model training and improve the positioning accu-racy of bounding box regression.Experimental results show that compared with the original model,the ESH-YOLO model im-proves the precision,recall rate,mAP@0.5 and F1 score by 0.9 percentage points,3.6 percentage points,2.8 percentage points and 2.3 percentage points respectively,which can effectively reduce the missed detection and false detection of small targets in complex scenes.The application of this algorithm in complex construction sites can accurately detect whether workers wear safety helmets,improve the wearing rate of safety helmets,so as to ensure the life safety of workers.关键词
安全帽佩戴检测/YOLOv8/小目标检测/Star_CAA/ELA_HSPAN/PIoUv2Key words
helmet wearing detection/YOLOv8/small target detection/Star_CAA/ELA_HSPAN/PIoUv2分类
信息技术与安全科学引用本文复制引用
夏争乐,艾菊梅..基于ESH-YOLO的小目标安全帽检测算法[J].计算机与现代化,2026,(3):95-101,7.基金项目
江西省放射性地学大数据技术工程实验室开放基金资助项目(JELRGBDT201805) （JELRGBDT201805）
