• Stars
    star
    250
  • Rank 158,658 (Top 4 %)
  • Language
    Python
  • License
    GNU General Publi...
  • Created over 5 years ago
  • Updated 9 months ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

SwiftNet

Source code to reproduce results from

Steps to reproduce

Install requirements

  • Python 3.7+
pip install -r requirements.txt

Download Cityscapes

From https://www.cityscapes-dataset.com/downloads/ download:

  • leftImg8bit_trainvaltest.zip (11GB)
  • gtFine_trainvaltest.zip (241MB)

Either download and extract to datasets/ or create a symbolic link datasets/Cityscapes Expected dataset structure for Cityscapes is:

labels/
    train/
        aachen/
            aachen_000000_000019.png
            ...
        ...
    val/
        ...
rgb/
    train/
        aachen/
            aachen_000000_000019.png
            ...
        ...
    val/
        ...

Evaluate

Pre-trained Cityscapes models available
  • Download and extract to weights directory.

Set evaluating = True inside config file (eg. configs/rn18_single_scale.py) and run:

python eval.py configs/rn18_single_scale.py

Train

python train.py configs/rn18_single_scale.py --store_dir=/path/to/store/experiments