tingyuz / keras-icnet

Keras implementation of Real-Time Semantic Segmentation on High-Resolution Images

Keras-ICNet

[paper]

Keras implementation of Real-Time Semantic Segmentation on High-Resolution Images. Training in progress!

Requisites

• Python 3.6.3
• Keras 2.1.1 with Tensorflow backend
• A dataset, such as Cityscapes or Mapillary (Mapillary was used in this case).

Train

Issue ./train --help for options to start a training session, default arguments should work out-of-the-box.

You need to place the dataset following the next directory convention:

.
├── mapillary                   
|   ├── training
|   |   ├── images             # Contains the input images
|   |   └── instances          # Contains the target labels
|   ├── validation
|   |   ├── images
|   |   └── instances
|   └── testing
|   |   └── images

These are the results of training for 300 epochs ./train --epochs 300

Training

Validation

Test

Issue ./test --help for options to start a testing session, default arguments should work out-of-the-box.

Output examples

TODO

• Perform class weighting