StyleGAN2.pytorch

StyleGAN2.pytorch copied to clipboard

pytorch version is very very import！！！ I test some repo for re-implement stylegan-v2 but all failed

and then they never update。。。。or fixed bugs

I wish you code will working and if you have any problem for help， you can say in readme or call me

Thank you for your attention to my work, in fact I am doing this recently. Regarding the implementation of stylegan2, I recommend https://github.com/genforce/mganprior to you. It includes a stylegan2 generator implemented in pure pytorch, but does not include the discriminator and training framework. The reason I recommend it is that its implementation is elegant and pure pytorch, so it is easy to understand and modify. I tried to add this implementation to the training framework, but now I can't train normally. I am looking for the reason. If you have any good suggestions, please let me know.

Best regard, @huangzh13

thank you for your recommend, I will check it,

About training, https://github.com/lucidrains/stylegan2-pytorch this repo seems to get good result , and someone can get too, but I failed! This training process is much more different from NVlabs, but maybe it work, you can try it