How can I train MaskGAN for another dataset?

switchablenorms / CelebAMask-HQ

A large-scale face dataset for face parsing, recognition, generation and editing.

2.05k stars 343 forks source link

How can I train MaskGAN for another dataset? #30

Closed ksoy0128 closed 4 years ago

ksoy0128 commented 4 years ago

Hi,

Thanks for sharing your code! MaskGAN is very interesting project. I tried to train MaskGAN for another dataset. I know that overall training pipeline is divided 2 steps. (Stage-1, Stage-2) However, I don't know how can i get pretrained Ga, Enc_VAE, Dec_VAE (Stage-1). If possible, can you tell me how to train MaskGAN for another dataset on this code?

Thank you.

steven413d commented 4 years ago

Hi, Thank for your interest in our project:) Sorry that we have not released the training code and pre-trained models yet. This code is only for inference.