big-sleep icon indicating copy to clipboard operation
big-sleep copied to clipboard

Implementing StyleGAN2

Open ShivanshuPurohit opened this issue 5 years ago • 4 comments

How should I go about to replace BigGAN with your StyleGAN2 package?

ShivanshuPurohit avatar Jan 20 '21 12:01 ShivanshuPurohit

what is needed is pretrained weights on a diverse set of 512x512 images (maybe Imagenet) for pytorch. If you can find that, it is doable!

lucidrains avatar Jan 20 '21 15:01 lucidrains

how to use with the stylegan2-ffhq-config-f.pkl? just download the pkl and replace it? if so, what line should I change to change biggan with stylegan2-ffhq-config-f.pkl?

molo32 avatar Jan 24 '21 04:01 molo32

Found this ImageNet-trained StyleGAN2, but unfortunately not for PyTorch: https://battle.shawwn.com/sdc/stylegan2-imagenet-512/model.ckpt-533504.pkl

Found via this tweet: https://twitter.com/theshawwn/status/1244081581347598341

Seems to be created using this NVIDIA-based TensorFlow repo: https://github.com/shawwn/stylegan2/tree/working

walmsley avatar Feb 07 '21 19:02 walmsley

any news on this?

illtellyoulater avatar Apr 17 '22 18:04 illtellyoulater