Detection Algorithm for Protective Equipment of Railway Workers Based on Improved YOLOv5

Ming-ming HOU , Xiao-chuan NIU , Ke-xun WANG

Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (23) : 9859 -9865.

PDF
Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (23) : 9859-9865 DOI: 10.12404/j.issn.1671-1815.2405697
research-article
Detection Algorithm for Protective Equipment of Railway Workers Based on Improved YOLOv5
Author information +
History +
PDF

Abstract

Currently, railway maintenance safety accidents occur frequently, and existing railway safety supervision schemes are inefficient and difficult to effectively ensure the safe conduct of maintenance activities. A protection equipment detection scheme for railway maintenance workers based on the improved YOLOv5 model was proposed, which realized real-time detection of safety helmets and reflective vests worn by railway maintenance personnel. To address issues such as missed detections, false positives, and inaccurate detections caused by the complex background of railway scenes, a spatial reconstruction unit (SRU) and an efficient channel attention(ECA) module was introduced. These modules were employed to enhance the network's feature expression ability in both the spatial and channel dimensions, effectively filtering out useful features and removing redundant ones. Additionally, weighted intersection over union (W-IOU) loss function was introduced to reduce the impact of low-quality samples on training, thereby improving the convergence ability of model. Experimental validation result shows that the model achieves the average accuracy of 94.3%, the precision of 94.9%, and the recall of 94.7%, which represent the improvements of 2.2%, 2.5%, and 1.6%, respectively, over the baseline network. The inference speed reaches 34.5 FPS, allowing for real-time detection in actual maintenance environments.

Keywords

YOLOv5 / engineering safety / deep learning / object detection

Cite this article

Download citation ▾
Ming-ming HOU, Xiao-chuan NIU, Ke-xun WANG. Detection Algorithm for Protective Equipment of Railway Workers Based on Improved YOLOv5. Science Technology and Engineering, 2025, 25(23): 9859-9865 DOI:10.12404/j.issn.1671-1815.2405697

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (10989KB)

18

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉