/Segnet-pytorch

Pytorch Implementation of Segnet for the LDC dataset.

Primary LanguagePython

Segnet-pytorch

Pytorch Implementation of Segnet for the LDC dataset.

Code inspired by @Sayan98

Usage: python train.py --data_root Data --train_path train.txt --img_dir Train --mask_dir Train_annot --save_dir Output --checkpoint model_best.pth --gpu 1