adityakumarakash/pytorch_segnet

Implementation of segnet in pytorch

pytorch_segnet

Adapted from https://github.com/meetshah1995/pytorch-semseg, https://github.com/trypag/pytorch-unet-segnet/blob/master/segnet.py

Requirements

• pytorch
• torchvision
• scipy
• tqdm
• tensorboardX

To train the model :

python train.py [-h] [--config [CONFIG]]

--config                Configuration file to use

To test the model :

python test.py [-h] [--config [CONFIG]]

--config                Configuration file to use