search

Zhendong-Wang / Diffusion-GAN

Official PyTorch implementation for paper: Diffusion-GAN: Training GANs with Diffusion
MIT License
626 stars 67 forks source link

How to apply DIffusion-GAN to StyleGAN3 #5

Open thuanz123 opened 2 years ago

thuanz123 commented 2 years ago

Hi, I want to ask what change I need to make to apply Diffusion-GAN to the StyleGAN3 repo ?

Zhendong-Wang commented 2 years ago

Hi there,

Thanks for your interest. The change could be quite simple. You could copy the diffusion.py from our file to replace the original augment.py and also replace the discriminator in StyleGAN3 with our discriminator. These are the two main steps for the move. You also need to modify some argument names and input timestep into the discriminator, but I think they can be easily debugged :).

gminorcoles commented 1 year ago

Hi, I m looking to use your diffusion process to improve an existing GAN using the plugin approach . I don't see diffusion.py in the repo here? Perhaps it has a different name? Thanks

Zhendong-Wang commented 1 year ago

https://github.com/Zhendong-Wang/Diffusion-GAN/blob/main/diffusion-stylegan2/training/diffusion.py

Liu-Yin commented 9 months ago

Hello there! Just wondering how to replace the discriminator in StyleGAN3 with your discriminator ? It would be grateful if you could tell me! Thx a lot!!!