search

kazuto1011 / deeplab-pytorch

PyTorch re-implementation of DeepLab v2 on COCO-Stuff / PASCAL VOC datasets
MIT License
1.09k stars 282 forks source link

How to train on the coco-stuff datasets? #115

Closed xiaoningpi closed 1 year ago

xiaoningpi commented 1 year ago

I want to train on the COCO-Stuff dataset. How can I do that

kazuto1011 commented 1 year ago

Please follow this setup instruction and run python main.py train --config-path configs/cocostuff164k.yaml. A pre-trained model is also available.