Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add "Progressive Growing of GANs" (ProGAN) model #1105

Open
wants to merge 19 commits into
base: master
Choose a base branch
from

Conversation

seovchinnikov
Copy link

@seovchinnikov seovchinnikov commented Jul 25, 2020

I've decided to add a new model to the framework - an implementation of the paper "Progressive Growing of GANs": https://arxiv.org/abs/1710.10196.
It's basically a GAN that generates images from a random noise with a support of high resolution images by training G and D nets progressively. I've decided to post a PR because found it easily to implement it inside this repo's framework so it fits it very well (despite the name of repo :)

Model requires dataset of type dataset_mode='single', generator netG='progan', discriminator netD='progan'.
Please note that opt.crop_size (default 256) == 4 * 2 ** opt.max_steps (default max_steps is 6).
ngf and ndf controlls dimensions of the backbone (128-512).
Network G is a master-generator (accumulates weights for eval) and network C (stands for current) is a
current trainable generator.

See also:
    https://github.com/tkarras/progressive_growing_of_gans
    https://github.com/odegeasslbc/Progressive-GAN-pytorch

Please note that PR contains changes from PR #1090 with NVIDIA apex because larger batch sizes are prefered.

@junyanz
Copy link
Owner

junyanz commented Jul 25, 2020

Thanks for your contribution. Have you been able to reproduce the FID number on the ProgGAN datasets?

@seovchinnikov
Copy link
Author

seovchinnikov commented Aug 11, 2020

Sorry for a late response, I've only tested it on our private dataset, need help to evaluate it on Celeba

@aashana94
Copy link

I am trying to run metrics using the original implementation of Progressive Growing of GAN by tkarras but I am not able to, has anyone any luck? or can help me with it?

https://github.com/tkarras/progressive_growing_of_gans

@ridhachahed
Copy link

Hey ! You use 2 generators with one being the running average of the other and set adam parameter beta1 to 0. Why this choice ? Why not use the same generator for training and testing ?

@maxmax1992
Copy link

May I ask why are you in the ProGanModel doing generation from torch.randn and not from source Image A? It seems also that you don't even save the image A as input, this contradicts with the whole cycle consistency of the CycleGan as there's no loss between source image A and reconstruction of A.

@junyanz
Copy link
Owner

junyanz commented Sep 10, 2021

We will leave it as is. It might introduce too many changes to add a new GAN to the current repo. But if anyone is interested in progressive gans, you can refer to this PR.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants