co-mod-gan icon indicating copy to clipboard operation
co-mod-gan copied to clipboard

Reproducing COCO-stuff

Open sigeisler opened this issue 3 years ago • 1 comments

Hi,

thanks for sharing this great code base including pre-trained models! However, I would like to train the COCO-stuff model from scratch. Can you give me some instructions on what to do? I just want to make sure that I do not miss sth.

Thanks, Simon

sigeisler avatar Sep 09 '21 12:09 sigeisler

In addition to our pix2pix mode, you need to add an embedding layer to the input label map. When processing data, also need to make sure that the input label map is correctly resamped (using nearest interpolation)

zsyzzsoft avatar Sep 12 '21 05:09 zsyzzsoft