Implementation of various models for Semantic Segmentation.
- FCN
- Segnet_VGG16
- Segnet with different pretrained networks
pip install -r requirements.txt
Currently supported loaders are only VOC PASCAL.
You can find all data here:
Dataloader requires path to data. This can be provided through config.yaml
.
You can for example run - batch of size 6 takes approximately 9GB of gpu memory
python train.py --n_epoch 30 --batch_size 6
This repo was inspired by various other repositories and some parts of code come directly from them. So I list them all here: (Thank you)