Abstract:Addressing the limitations of current railway track foreign object detection techniques, which suffer from inadequate real-time performance and diminished accuracy in detecting small objects, this paper introduces an innovative vision-based perception methodology harnessing the power of deep learning. Central to this approach is the construction of a railway boundary model utilizing a sophisticated track detection method, along with an enhanced UNet semantic segmentation network to achieve autonomous segmentation of diverse track categories. By employing equal interval division and row-by-row traversal, critical track feature points are precisely extracted, and the track linear equation is derived through the least squares method, thus establishing an accurate railway boundary model. We optimized the YOLOv5s detection model in four aspects: incorporating the SE attention mechanism into the Neck network layer to enhance the model's feature extraction capabilities, adding a prediction layer to improve the detection performance for small objects, proposing a linear size scaling method to obtain suitable anchor boxes, and utilizing Inner-IoU to refine the boundary regression loss function, thereby increasing the positioning accuracy of the bounding boxes. We conducted a detection accuracy validation for railway track foreign object intrusion using a self-constructed image dataset. The results indicate that the proposed semantic segmentation model achieved an MIoU of 91.8%, representing a 3.9% improvement over the previous model, effectively segmenting railway tracks. Additionally, the optimized detection model could effectively detect foreign object intrusions on the tracks, reducing missed and false alarms and achieving a 7.4% increase in the mean average precision (IoU = 0.5) compared to the original YOLOv5s model. The model exhibits strong generalization capabilities in scenarios involving small objects. This proposed approach represents an effective exploration of deep learning techniques for railway track foreign object intrusion detection, suitable for use in complex environments to ensure the operational safety of rail lines.

Automatic Obstacle Detection Method for the Train Based on Deep Learning

Obstacle detection of rail transit based on deep learning

Obstacle Detection in Dangerous Railway Track Areas by a Convolutional Neural Network

Railway obstacle intrusion warning mechanism integrating YOLO-based detection and risk assessment

An Improved Deep Learning Algorithm for Obstacle Detection in Complex Rail Transit Environments

Unsupervised Deep Learning-Based Point Cloud Detection for Railway Foreign Object Intrusion

Deep-Learning-Based Detection of Obstacles in Transit on Trams

Robust Artificial Intelligence-Aided Multimodal Rail-Obstacle Detection Method by Rail Track Topology Reconstruction

Research and Application of the Obstacle Avoidance System for High-Speed Railway Tunnel Lining Inspection Train Based on Integrated 3D LiDAR and 2D Camera Machine Vision Technology

An Improved Single-Stage Convolutional Neural Network for Rail Transit Obstacle Detection

Railway Obstacle Detection using Onboard Forward-Viewing Camera

Research on the Method of Foreign Object Detection for Railway Tracks Based on Deep Learning

A Real-Time Method for Railway Track Detection and 3D Fitting Based on Camera and LiDAR Fusion Sensing

An Efficient Network for Obstacle Detection in Rail Transit Based on Multi-Task Learning

Autonomous Railway Traffic Object Detection Using Feature-Enhanced Single-Shot Detector

Multi-Sensor Fusion Perception System in Train

Application of Lightweight Railway Transit Object Detector

Urban rail transit obstacle detection based on Improved R-CNN

Subway Obstacle Perception and Identification Method Based on Cloud Edge Collaboration

Foreign object detection in urban rail transit based on deep differentiation segmentation neural network

Rail Transit Obstacle Detection Based on Improved CNN