Computer Science ›› 2025, Vol. 52 ›› Issue (11A): 250100047-7.doi: 10.11896/jsjkx.250100047

• Image Processing & Multimedia Technology • Previous Articles     Next Articles

Real-time Helmet Detection Algorithm for Roadway Engineering Construction Based on UAV Visual Inspection

WEN Ming1, WU Xingtang2, SHANG Yuhao2, ZHEN Jian3, YU Fucai1   

  1. 1 Beijing Academy of Emergency Management Science and Technology,Beijing 101101,China
    2 School of Control and Computer Engineering,North China Electric Power University,Beijing 102206,China
    3 Yangtze Delta Institute of Infrastructure,Hangzhou 310005,China
  • Online:2025-11-15 Published:2025-11-10
  • Supported by:
    Beijing Natural Science Foundation of China(9244040).

Abstract: To ensure the safety of highway engineering construction personnel and reduce safety risks during the construction process,real-time detection of helmet usage has become a critical safety supervision method.Highway projects are characterized by numerous,long,and wide construction sites,often involving complex terrains such as mountain ranges and rivers.Traditional fixed-camera coverage has limitations and high costs.Drones,as flexible,low-cost,and highly visible image acquisition tools,can effectively address these challenges,especially in high-risk areas that are difficult to cover with traditional methods.This paper proposes a real-time helmet detection algorithm based on an improved eXtended Difference of Gaussians(XDOG) and YOLOv5,aiming to solve the issues of misdetection and missed detection under variable lighting conditions,scale,and shape changes in images captured by drones.In complex construction environments,the features of safety helmets are often hard to distinguish from backgrounds or other objects.The XDOG module is introduced to enhance edge information in images,thereby highlighting the structural and detailed features of helmets to be detected.The difference-of-Gaussians results are further normalized and non-linearly activated to eliminate the effects of lighting variation and noise interference in construction environments.To ensure compatibility with the YOLOv5 network,the algorithm uses a 1×1 convolution layer to adjust the number of channels in the enhanced feature maps,and a residual connection is used to fuse the enhanced feature maps with the input image,thereby improving the robustness and accuracy of the network.Experimental results show that compared to traditional YOLOv5 and YOLOx models,the XDOG-YOLOv5 significantly improves detection accuracy,with notable gains in mAP@50 and mAP@50-95,demonstrating its effectiveness in real-time helmet detection for construction personnel.

Key words: Safety helmet detection, YOLOv5, Difference of Gaussians, Roadway engineering, Unmanned aerial vehicle

CLC Number: 

  • TP391
[1]YANG Y,PEI Z Z,CHEN Z,et al.Study on freeway traffic safety evaluation based on improved system engineering decision theory[J].Journal of Beijing Jiaotong University,2022,46(3):34-48.
[2]FENG P,QIAN Y R,FAN Y Y,et al.Safety helmet detection al-gorithm based on the improved Cascade R-CNN[J].Microelectronics & Computer,2024,41(1):63-73.
[3]GUO W,FAN Y G,LI X Z,et al.Detection of Nonstandard Wearing of Safety Helmet Based on Improved YOLO[J].Ordnance Industry Automation,2024,43(5):33-36,42.
[4]WANG Z B,MA H,FENG J L,et al.Helmet Detection Algorithm Based on CE-YOLOv5s[J].Computer and Modernization,2024(4):55-59,98.
[5]ZHANG G P,ZHOU J Z,MA G C,et al.Lightweight safety helmet wearing detection algorithm of improved YOLOv8[J].Electronic Measurement Technology,2024,47(17):147-154.
[6]LI Z S,XIE W Q,ZHANG L Z,et al.TowardEfficient Safety Helmet Detection Based on YoloV5 With Hierarchical Positive Sample Selection and Box Density Filtering[J].IEEE Transactions on Instrumentation and Measurement,2022,71:1-14.
[7]WAN J C,LIANG B C.Design and Implementation of an Intelligent Motorcycle Helmet for Large Vehicle Approach Intimation[J].IEEE Sensors Journal,2019,19(10):3882-3892.
[8]PATEL K,PATEL V,PRAJAPATI V,et al.Safety Helmet De-tection Using YOLO V8[C]//2023 3rd International Confe-rence on Pervasive Computing and Social Networking(ICPCSN).2023:22-26.
[9]ZHAO L,ZHANG D,LIU Y,et al.Improved YOLOv5s Network for Multi-scale safety Helmet Detection [C]//2022 11th International Conference on Communications,Circuits and Systems(ICCCAS).2022:262-266.
[10]WANG B,XIONG H,LIU L.Safety helmet wearing recognition based on improved YOLOv4 algorithm[C]//2022 IEEE 6th Information Technology and Mechatronics Engineering Conference(ITOEC).2022:1732-1736.
[11]AI X,CHEN C,WANG Y.Multi-Scene Safety Helmet Detection with Multi-Scale Spatial Attention Feature[C]//2021 7th IEEE International Conference on Network Intelligence and Di-gital Content(IC-NIDC).2021:21-25.
[12]CHEN L,MAO Y,ZHANG H.Improved the Detection Algo-rithm of Safety Helmet Wearing Based on YOLOv8[C]//2023 3rd International Conference on Electronic Information Engineering and Computer Communication(EIECC).2023:1-4.
[13]QU H,LOU S,WANG X.Safety Helmet Wearing Detection Algorithm Based on Improved Yolov5s[C]//2023 5th International Conference on Artificial Intelligence and Computer Applications(ICAICA).2023:37-40.
[14]NING S,HAN X.An improved YOLOv8-based safety helmetwearing detection algorithm[C]//2024 7th International Conference on Computer Information Science and Application Technology(CISAT).2024:75-79.
[15]XU K,LI X Z,ZHANG L,et al.Safety Helmet Wearing Detection Algorithm for Distribution Network Construction in Natural Scenarios[J].Computer Engineering and Applications,2024,60(8):320-328.
[16]ZHENG C W,LIN H.YOLOv5 Helmet Wearing DetectionMethod Based on Swin Transformer[J].Computer Measurement & Control,2023,31(3):15-21.
[17]DU Q,YANG S J,GUO Q P.Intelligent detection method of working personnel wearing safety helmets in underground mine[J].Journal of Mine Automation,2023,49(7):134-140.
[18]LI Z F,FENG S Y,GUO J,et al.Lightweight safety helmet wearing detection fusing coordinate attention and multiscale feature[J].Journal of Mine Automation,2023,49(11):151-159.
[1] CHEN Shijia, YE Jianyuan, GONG Xuan, ZENG Kang, NI Pengcheng. Aircraft Landing Gear Safety Pin Detection Algorithm Based on Improved YOlOv5s [J]. Computer Science, 2025, 52(6A): 240400189-7.
[2] DING Xuxing, ZHOU Xueding, QIAN Qiang, REN Yueyue, FENG Youhong. High-precision and Real-time Detection Algorithm for Photovoltaic Glass Edge Defects Based onFeature Reuse and Cheap Operation [J]. Computer Science, 2025, 52(6A): 240400146-10.
[3] MENG Dongyue, HUANG Yuchuan, HAN Guoxiang, LI Hongchen, WANG Pengfei. Research on Emergency Rescue Quadcopter UAV Safety Control Based on Feedforward PID [J]. Computer Science, 2025, 52(11A): 241200203-9.
[4] WEN Haolin, LIANG Xin, CHEN Tong, LI Yuqi. UAV Logistics Network Planning Method Considering Demand and Range [J]. Computer Science, 2025, 52(11A): 250200042-5.
[5] ZHANG Meng, QIAO Jinlan. Research on Individual Unmanned Aerial Vehicles Identification Technology Based on Voiceprint Characteristics [J]. Computer Science, 2025, 52(11): 444-451.
[6] CHEN Chongyang, PENG Li, YANG Jielong. UAV Small Object Detection Algorithm Based on Feature Enhancement and Context Fusion [J]. Computer Science, 2025, 52(11): 131-140.
[7] XU Jinlong, GUI Zhonghua, LI Jia'nan, LI Yingying, HAN Lin. FP8 Quantization and Inference Memory Optimization Based on MLIR [J]. Computer Science, 2024, 51(9): 112-120.
[8] HU Pengfei, WANG Youguo, ZHAI Qiqing, YAN Jun, BAI Quan. Night Vehicle Detection Algorithm Based on YOLOv5s and Bistable Stochastic Resonance [J]. Computer Science, 2024, 51(9): 173-181.
[9] LI Yuanxin, GUO Zhongfeng, YANG Junlin. Container Lock Hole Recognition Algorithm Based on Lightweight YOLOv5s [J]. Computer Science, 2024, 51(6A): 230900021-6.
[10] WANG Zhen, ZHOU Chao, FAN Yongwen, Shi Pengfei. Overview of Unmanned Aerial Vehicle Systems Security [J]. Computer Science, 2024, 51(6A): 230800086-6.
[11] MA Yinghong, LI Xu’nan, DONG Xu, JIAO Yi, CAI Wei, GUO Youguang. Fast Path Recovery Algorithm for Obstacle Avoidance Scenarios [J]. Computer Science, 2024, 51(6): 331-337.
[12] LU Yanfeng, WU Tao, LIU Chunsheng, YAN Kang, QU Yuben. Survey of UAV-assisted Energy-Efficient Edge Federated Learning [J]. Computer Science, 2024, 51(4): 270-279.
[13] PANG Yuxiang, CHEN Zemao. Security Scheme of UAV Flight Control Based on Attribute Access Control Policy [J]. Computer Science, 2024, 51(4): 366-372.
[14] ZHANG Yang, XIA Ying. Object Detection Method with Multi-scale Feature Fusion for Remote Sensing Images [J]. Computer Science, 2024, 51(3): 165-173.
[15] WANG Xinlong, LIN Bing, CHEN Xing. Computation Offloading with Wardrop Routing Game in Multi-UAV-aided MEC Environment [J]. Computer Science, 2024, 51(3): 309-316.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!