UAV-assisted Railway Track Segmentation based on Convolutional Neural Networks
Abdelhamid Mammeri, Abdul Jabbar Siddiqui, Yiheng Zhao
Abstract
In the railway sector, track inspections are regularly needed to monitor the track conditions for potential hazards in order to ensure safety and security of life and property. Recently, conducting infrastructure inspections and monitoring using UAVs has gained attention in various industries including the railways. The rapid development of advanced deep learning and machine vision techniques have given rise to automated railway hazard detection systems based on UAV-based imagery. A major task in such systems is to localize or segment the railway tracks in UAV-based images. This paper investigates the effectiveness of a fully convolutional encoder-decoder type segmentation network called U-Net for the task of segmenting rail track regions from UAV-based images. Through experimental evaluations using a proprietary real-world dataset, we demonstrate U-Net's effectiveness in terms of mean Intersection over Union (IoU). Such methods of rail track segmentation are particularly useful in applications such as automated UAV navigation along rail tracks.