Visual monitoring of Railway obstacles using deep learning-based methods

Document Type : Scientific - Research

Authors

1 Associate Professor, Control & Signalling, School of Railway Engineering, Iran University of Science & Technology (IUST)

2 Control and Signaling, School of Railway Engineering, Iran University of Science and Technology

Abstract

One significant aspect contributing to the overall improvement of railway operations and safety is the detection of obstacles along the tracks. This advancement not only enhances the efficiency of rail transport but also ensures greater accuracy and reliability. Thanks to recent advancements in artificial intelligence technologies, obstacle detection and self-driving methods have witnessed remarkable progress.in this project, a structure has been designed that enables the real-time detection of obstacles on the train track with a balance between accuracy and volume of calculations. The proposed method is to identify the train path and check the presence of obstacles in this area., three segmentation models PAN+SE-ResNet, PAN+EfficientNet and PAN+NF-Net were designed and trained and validated by a set of color images. results show that the PAN+NF-Net rail segmentation model is more accurate than the other two models with a few tenths of a percent difference. Meanwhile, the computing load of the PAN+EfficientNet model is about half and one-fifth of the computational load of the PAN+NF-Net and PAN+SE-ResNet models , and it has a higher processing speed. In order to design the object recognition model, the fifth and seventh versions of the YOLO algorithm were trained by a ًأ‌ dataset. The results show that the accuracy of the YOLOV7 model in detecting obstacles is 17.6 percent higher than the accuracy of the YOLOV5 model. Finally, obstacle detection was done by checking the existence of overlap between the mask of the train's path and the bounding boxes predicted by the object detection model.

Keywords

Main Subjects



Articles in Press, Accepted Manuscript
Available Online from 30 December 2023
  • Receive Date: 12 July 2023
  • Revise Date: 12 October 2023
  • Accept Date: 16 November 2023