Railway Switch Classification Using Deep Neural Networks (2023)

Abstract

Railway switches represent the mechanism which slightly adjusts the rail blades at the intersection of two rail
tracks in order to allow trains to exchange their routes. Ensuring that the switches are correctly set represents
a critical task. If switches are not correctly set, they may cause delays in train schedules or even loss of lives.
In this paper we propose an approach for classifying switches using various deep learning architectures with
a small number of parameters. We exploit various input modalities including: grayscale images, black and
white binary masks and a concatenated representation consisting of both. The experiments are conducted on
RailSem19, the most comprehensive dataset for the task of switch classification, using both fine-tuned models
and modelstrained from scratch. The switch bounding boxes from the dataset are pre-processed by introducing
three hyper-parameters over the boxes, improving the models performance. We manage to achieve an overall
accuracy of up to 96% in a ternary multi-class classification setting where our model is able to distinguish
between images containing left, right or no switches at all. The results for the left and right switch classes are
compared with two other existing approaches from the literature. We obtain competitive results using deep
neural networks with considerably fewer learnable parameters than the ones from the literature.

Citare

@Inproceedings{Alexandrescu2023RailwaySC,
 author = {Andrei-Robert Alexandrescu and Alexandru Manole and L. Dioşan},
 booktitle = {VISIGRAPP},
 title = {Railway Switch Classification Using Deep Neural Networks},
 year = {2023}
}

Leave a Reply

Your email address will not be published. Required fields are marked *