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.

Multi-Scale Rotated Bounding Box-Based Deep Learning Method For Electric Railway Detection

A Multi-scale Detection Method for Dropper States in High-speed Railway Contact Network Based on RefineDet Network and Hough Transform

An Improved Faster R-CNN for Devices Detection in Railway 4C System

A Lightweight Framework for Obstacle Detection in the Railway Image Based on Fast Region Proposal and Improved YOLO-Tiny Network

Multialgorithm Fusion Image Processing for High Speed Railway Dropper Failure–Defect Detection

Defect Recognition of Insulators on Catenary Via Multi-Oriented Detection and Deep Metric Learning

Rotating object detection in remote-sensing environment

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

Railway insulator defect detection with deep convolutional neural networks

A Hybrid Deep Learning Based Framework for Component Defect Detection of Moving Trains

Application of Lightweight Railway Transit Object Detector

Intelligent Image-Based Railway Inspection System Using Deep Learning-Based Object Detection and Weber Contrast-Based Image Comparison

Hybrid deep learning architecture for rail surface segmentation and surface defect detection

A Study on Railway Surface Defects Detection Based on Machine Vision

Research on deep learning method for rail surface defect detection

An object detection method for catenary component images based on improved Faster R-CNN

An Effective Railway Intrusion Detection Method Using Dynamic Intrusion Region and Lightweight Neural Network

Obstacle Detection Method of Underground Electric Locomotive Rail Based on Instance Segmentation

Real-Time Accurate Deep Learning-Based Edge Detection for 3-D Pantograph Pose Status Inspection

Deep Learning for the Detection and Recognition of Rail Defects in Ultrasound B-Scan Images

CPS-Det: An Anchor-Free Based Rotation Detector for Ship Detection