Target Detection of Safety Protective Gear Using the Improved YOLOv5

被引:0
作者
Liu, Hao [1 ]
Qin, Xue [1 ]
机构
[1] Guizhou Univ, Coll Big Data & Informat Engn, Guiyang, Peoples R China
来源
2024 5TH INTERNATIONAL CONFERENCE ON COMPUTERS AND ARTIFICIAL INTELLIGENCE TECHNOLOGY, CAIT | 2024年
关键词
Object detection; Construction industry; Safety devices; Computer vision; Deep learning; MOTIVATION; BELT;
D O I
10.1109/CAIT64506.2024.10962947
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In high-risk railway construction, personal protective equipment monitoring is critical but challenging due to small and frequently obstructed targets. We propose YOLO-EA, an innovative model that enhances safety measure detection by integrating ECA into its backbone's convolutional layers, improving discernment of minuscule objects like hardhats. YOLO-EA further refines target recognition under occlusion by replacing GIoU with EIoU loss. YOLO-EA's effectiveness was empirically substantiated using a dataset derived from real-world railway construction site surveillance footage. It outperforms YOLOv5, achieving 98.9% precision and 94.7% recall, up 2.5% and 0.5% respectively, while maintaining real-time performance at 70.774 fps. This highly efficient and precise YOLO-EA holds great promise for practical application in intricate construction scenarios, enforcing stringent safety compliance during complex railway construction projects.
引用
收藏
页码:6 / 13
页数:8
相关论文
共 18 条
[1]   Half and Full Helmet Wearing Detection in Thailand using Haar Like Feature and Circle Hough Transform on Image Processing [J].
Doungmala, Pathasu ;
Klubsuwan, Katanyoo .
2016 IEEE INTERNATIONAL CONFERENCE ON COMPUTER AND INFORMATION TECHNOLOGY (CIT), 2016, :611-614
[2]  
Fangbo Zhou, 2021, Proceedings of the 2021 IEEE International Conference on Power Electronics, Computer Applications (ICPECA), P6, DOI 10.1109/ICPECA51329.2021.9362711
[3]   Fast R-CNN [J].
Girshick, Ross .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1440-1448
[4]   Rich feature hierarchies for accurate object detection and semantic segmentation [J].
Girshick, Ross ;
Donahue, Jeff ;
Darrell, Trevor ;
Malik, Jitendra .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587
[5]   Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2015, 37 (09) :1904-1916
[6]   Understanding China's Belt & Road Initiative: Motivation, framework and assessment [J].
Huang, Yiping .
CHINA ECONOMIC REVIEW, 2016, 40 :314-321
[7]  
ipaf.org, IPAF Global Safety Report
[8]   DWCA-YOLOv5: An Improve Single Shot Detector for Safety Helmet Detection [J].
Jin, Zhang ;
Qu, Peiqi ;
Sun, Cheng ;
Luo, Meng ;
Gui, Yan ;
Zhang, Jianming ;
Liu, Hong .
JOURNAL OF SENSORS, 2021, 2021
[9]  
kaggle.com, About us
[10]   SSD: Single Shot MultiBox Detector [J].
Liu, Wei ;
Anguelov, Dragomir ;
Erhan, Dumitru ;
Szegedy, Christian ;
Reed, Scott ;
Fu, Cheng-Yang ;
Berg, Alexander C. .
COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 :21-37