stylegan2-pytorch
stylegan2-pytorch copied to clipboard
Add easy link for converted weight file.
Running the convert_weight.py script has lots of dependency issues (e.g the original stylegan2 uses Tensorflow 1.x). Hopefully this makes the process of getting the script to work less painful for others, who just want the pretrained model.
For completeness, the generator can then be loaded with
from model import Generator
G = Generator(1024, 512, 8)
G.load_state_dict(torch.load("stylegan2-ffhq-config-f.pt")['g_ema'])
Hi ArthurConmy,
This really helps. I have been working with the weight convert and I failed to get it work.
If I am understanding correctly, you have made the convertion work and posed a link for the ffhq converted checkpoint. I am wondering if you could be kind enough to additionally convert the following 2 checkpoints and post a link so that I can download it:
stylegan2-church-config-f.pkl stylegan2-church-config-a.pkl
They are from the official pretrained network link provided in the repo stylegan2
This would help a lot!