kazuto1011 / deeplab-pytorch

PyTorch re-implementation of DeepLab v2 on COCO-Stuff / PASCAL VOC datasets

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to train on the coco-stuff datasets?

xiaoningpi opened this issue · comments

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

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