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
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