How to train BigGAN for image generation conditioned on text description?

ajbrock / BigGAN-PyTorch

The author's officially unofficial PyTorch BigGAN implementation.

MIT License

2.84k stars 471 forks source link

How to train BigGAN for image generation conditioned on text description? #95

Open jaygala24 opened 2 years ago

jaygala24 commented 2 years ago

Hey everyone,

I was exploring the landscape for text to image generation using generative adversarial networks (GANs). I was wondering if anyone has previously tried training BigGAN for generating images conditioned on the text description. If we just change the nn.Embedding to nn.Linear in place of class condition embedding then will it work. If anyone can share some pointers regarding modifying BigGAN for text-to-image then it would be really helpful.

Thanks!

zhang373 commented 8 months ago

Maybe you should check GigaGAN