how to train encnet on cityscape dataset

zhanghang1989 / PyTorch-Encoding

A CV toolkit for my papers.

https://hangzhang.org/PyTorch-Encoding/

MIT License

2.04k stars 450 forks source link

how to train encnet on cityscape dataset #118

Closed gjm441 closed 5 years ago

gjm441 commented 5 years ago

Hi zhanghang, great work for sharing. I have trained VOCAug dataset and then finetune on VOC dataset, get mIOU 0.8506 on val. Have you provided coco pretrained encnet model in modelzoo?And how to train encnet on cityscape dataset? I hope you can give some suggestions,thanks

zhanghang1989 commented 5 years ago

Hi, I haven't tried EncNet on Cityscapes dataset. I expect you may need to enlarge the crop size to fix the high resolution setting in that dataset (probably crop_size = 768).

gjm441 commented 5 years ago

Thanks for kindly reply.Can you provide coco trained encnet model?

zhanghang1989 commented 5 years ago

Will do. I will let you know when it is ready.