orsic / swiftnet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

License:GNU General Public License v3.0


Languages

Language:Python 97.5%Language:C++ 2.2%Language:Shell 0.3%