Tramac / awesome-semantic-segmentation-pytorch

Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)
Apache License 2.0
2.85k stars 582 forks source link

How to train mscoco dataset? #104

Open ErinHoudini opened 4 years ago

ErinHoudini commented 4 years ago

Hi,thank you for your great job. I'm a new pytorch learner,I was wondering how to train mscoco dataset when model is fcn32s. The choice in '--dataset' in demo.py are ['pascal_voc/pascal_aug/ade20k/citys']. 'mscoco' is not included,And train.py also doesn't have mscoco in ''--dataset''. So how can I add mscoo dataset as a training dataset? I will be very appreciated if you can answer my questions.