Railways Segmentation

Track segmentation on high-resolution railway imagery

Two SwiftNet models segment the rails on 2048×1084 frames from a locomotive-mounted camera.

Try it

Choose one of the presets or upload an image

Results will appear here
Uploading…

Algorithm

Stack
  • Python 3.8
  • Kedro
  • MLFlow
  • Pytorch
  • LabelMe
Dataset
  • 23200 images
  • PNG/JPEG
  • Vector labeling
  • No augmentation
Training
  • SwiftNet
  • 50 epochs
  • 2x GTX1080TI
  • Full resolution

Model

Pipeline scheme

Publications

Railroad semantic segmentation on high-resolution images

Sergey Belyaev, Igor Popov, Vladislav Shubnikov, Pavel Popov, Ekaterina Boltenkova, Daniil Savchuk

2020 IEEE 23rd International Conference on Intelligent Transportation Systems (ITSC), 2020, pp. 1-6

doi: 10.1109/ITSC45102.2020.9294722

Download Learn more