stylegan2-pytorch icon indicating copy to clipboard operation
stylegan2-pytorch copied to clipboard

Add easy link for converted weight file.

Open ArthurConmy opened this issue 3 years ago • 2 comments

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.

ArthurConmy avatar Apr 22 '21 11:04 ArthurConmy

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'])

ArthurConmy avatar Apr 22 '21 13:04 ArthurConmy

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!

lizhihao98 avatar Apr 01 '22 04:04 lizhihao98