search

hubert0527 / COCO-GAN

COCO-GAN: Generation by Parts via Conditional Coordinating (ICCV 2019 oral)
https://hubert0527.github.io/COCO-GAN/
MIT License
268 stars 29 forks source link

sample script for sampling from trained models #3

Closed dribnet closed 5 years ago

dribnet commented 5 years ago

Hello- would it be possible to provide a sample script or perhaps some guidance on how one would sample from one of the pre-trained models?

hubert0527 commented 5 years ago

The simplest way is to simply run the training script, and the full images will be shown on the Tensorboard logs. I have never seen our model mode-collapses, so basically you can train it forever.

Another programmatic way is to get it by calling trainer.rand_sample_full_test(). It returns a batch of images as a numpy array.