Railway Foreign Object Intrusion Detection Using UAV Images and YOLO-UAT

Rapidly detecting foreign objects in the railway’s natural environment is crucial for safe railway operation and passenger safety. Traditional detection methods are limited by technology and environment and are costly and inefficient for large-scale detection. To improve the efficiency of...

Full description

Saved in:
Bibliographic Details
Main Authors: Yang Yang, Zhanhao Liu, Junming Chen, Haiming Gao, Tao Wang
Format: Article
Language:English
Published: IEEE 2025-01-01
Series:IEEE Access
Subjects:
Online Access:https://ieeexplore.ieee.org/document/10851273/
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1832576742032670720
author Yang Yang
Zhanhao Liu
Junming Chen
Haiming Gao
Tao Wang
author_facet Yang Yang
Zhanhao Liu
Junming Chen
Haiming Gao
Tao Wang
author_sort Yang Yang
collection DOAJ
description Rapidly detecting foreign objects in the railway&#x2019;s natural environment is crucial for safe railway operation and passenger safety. Traditional detection methods are limited by technology and environment and are costly and inefficient for large-scale detection. To improve the efficiency of foreign object intrusion detection in railways, this study proposes a YOLO-UAT method to detect foreign objects in railways effectively. First, a railway foreign object intrusion image dataset is created. Then the EfficientNet network is employed to replace the backbone extraction network of YOLOv5s, achieving a lightweight model that enhances detection speed. Secondly, a C<inline-formula> <tex-math notation="LaTeX">$3\_$ </tex-math></inline-formula> CBAM module is constructed to enhance feature extraction and enhance the model&#x2019;s detection ability for small-scale targets. Concurrently, the K-means++ algorithm is introduced to cluster the a priori frames, which improves the accuracy and convergence speed of the a priori frame clustering. YOLO-UAT reduces the number of parameters by 36% compared to the original YOLOv5s and mAP increased by 6.1% to 91.5%. Implemented on a Jetson Nano, it achieves a detection rate of 26.4 FPS. The experimental results demonstrate that the enhanced YOLOv5s model effectively enhances the accuracy and speed of foreign object encroachment detection in railroads while ensuring lightweightness, facilitating deployment, and aligning with the operational needs of railroads.
format Article
id doaj-art-23f6709ff63947a9898ccd713fbf2003
institution Kabale University
issn 2169-3536
language English
publishDate 2025-01-01
publisher IEEE
record_format Article
series IEEE Access
spelling doaj-art-23f6709ff63947a9898ccd713fbf20032025-01-31T00:01:54ZengIEEEIEEE Access2169-35362025-01-0113184981850910.1109/ACCESS.2025.353330410851273Railway Foreign Object Intrusion Detection Using UAV Images and YOLO-UATYang Yang0https://orcid.org/0000-0001-9257-4631Zhanhao Liu1https://orcid.org/0009-0000-4365-0386Junming Chen2https://orcid.org/0009-0002-8949-9620Haiming Gao3https://orcid.org/0009-0004-6369-9805Tao Wang4https://orcid.org/0009-0004-8817-313XSchool of Information Science and Engineering, Zhejiang Sci-Tech University, Hangzhou, ChinaSchool of Information Science and Engineering, Zhejiang Sci-Tech University, Hangzhou, ChinaSchool of Information Science and Engineering, Zhejiang Sci-Tech University, Hangzhou, ChinaSchool of Information Science and Engineering, Zhejiang Sci-Tech University, Hangzhou, ChinaSchool of Information Science and Engineering, Zhejiang Sci-Tech University, Hangzhou, ChinaRapidly detecting foreign objects in the railway&#x2019;s natural environment is crucial for safe railway operation and passenger safety. Traditional detection methods are limited by technology and environment and are costly and inefficient for large-scale detection. To improve the efficiency of foreign object intrusion detection in railways, this study proposes a YOLO-UAT method to detect foreign objects in railways effectively. First, a railway foreign object intrusion image dataset is created. Then the EfficientNet network is employed to replace the backbone extraction network of YOLOv5s, achieving a lightweight model that enhances detection speed. Secondly, a C<inline-formula> <tex-math notation="LaTeX">$3\_$ </tex-math></inline-formula> CBAM module is constructed to enhance feature extraction and enhance the model&#x2019;s detection ability for small-scale targets. Concurrently, the K-means++ algorithm is introduced to cluster the a priori frames, which improves the accuracy and convergence speed of the a priori frame clustering. YOLO-UAT reduces the number of parameters by 36% compared to the original YOLOv5s and mAP increased by 6.1% to 91.5%. Implemented on a Jetson Nano, it achieves a detection rate of 26.4 FPS. The experimental results demonstrate that the enhanced YOLOv5s model effectively enhances the accuracy and speed of foreign object encroachment detection in railroads while ensuring lightweightness, facilitating deployment, and aligning with the operational needs of railroads.https://ieeexplore.ieee.org/document/10851273/Object detectionUAVYOLOv5sEfficientNetlightweightCBAM
spellingShingle Yang Yang
Zhanhao Liu
Junming Chen
Haiming Gao
Tao Wang
Railway Foreign Object Intrusion Detection Using UAV Images and YOLO-UAT
IEEE Access
Object detection
UAV
YOLOv5s
EfficientNet
lightweight
CBAM
title Railway Foreign Object Intrusion Detection Using UAV Images and YOLO-UAT
title_full Railway Foreign Object Intrusion Detection Using UAV Images and YOLO-UAT
title_fullStr Railway Foreign Object Intrusion Detection Using UAV Images and YOLO-UAT
title_full_unstemmed Railway Foreign Object Intrusion Detection Using UAV Images and YOLO-UAT
title_short Railway Foreign Object Intrusion Detection Using UAV Images and YOLO-UAT
title_sort railway foreign object intrusion detection using uav images and yolo uat
topic Object detection
UAV
YOLOv5s
EfficientNet
lightweight
CBAM
url https://ieeexplore.ieee.org/document/10851273/
work_keys_str_mv AT yangyang railwayforeignobjectintrusiondetectionusinguavimagesandyolouat
AT zhanhaoliu railwayforeignobjectintrusiondetectionusinguavimagesandyolouat
AT junmingchen railwayforeignobjectintrusiondetectionusinguavimagesandyolouat
AT haiminggao railwayforeignobjectintrusiondetectionusinguavimagesandyolouat
AT taowang railwayforeignobjectintrusiondetectionusinguavimagesandyolouat