Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add easy link for converted weight file. #218

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Add easy link for converted weight file. #218

wants to merge 1 commit into from

Conversation

ArthurConmy
Copy link

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
Copy link
Author

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

@@ -45,6 +45,8 @@ For example, if you cloned repositories in ~/stylegan2 and downloaded stylegan2-

This will create converted stylegan2-ffhq-config-f.pt file.

Alternatively, the stylegan2-ffhq-config-f.pt can be found at https://drive.google.com/file/d/1EC34VzQ-7W6X1t9az-GEFOWX1uYPpW8w/view.

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

thanks

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

十分感谢

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@johndpope I can't access this url because I don't have permission, can you give me access?

@lizhihao98
Copy link

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!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants