search

rosinality / stylegan2-pytorch

Implementation of Analyzing and Improving the Image Quality of StyleGAN (StyleGAN 2) in PyTorch
MIT License
2.73k stars 620 forks source link

Convert weights from Stylegan2-ada-pytorch #206

Open dokluch opened 3 years ago

dokluch commented 3 years ago

Hi. Is it possible to convert weights from stylegan2-ada-pytorch (https://github.com/NVlabs/stylegan2-ada-pytorch) pickle file to pt with convert_weight tool?

rosinality commented 3 years ago

No, it is not possible currently.

ghost commented 3 years ago

No, it is not possible currently.

I hope it will be as well 👯

dvschultz commented 3 years ago

Justin Pinkney wrote one that works for me in 1024 and 512: https://github.com/dvschultz/stylegan2-ada-pytorch/blob/main/export_weights.py

rosinality commented 3 years ago

Wow!

ykk648 commented 3 years ago

Justin Pinkney wrote one that works for me in 1024 and 512: https://github.com/dvschultz/stylegan2-ada-pytorch/blob/main/export_weights.py

It works, but I found img generated by Stylegan2-ada-pytorch and Stylegan2-pytorch is different!!! The img generated by Stylegan2-pytorch has low quality， many strips on it.

CurtisASmith commented 3 years ago

Justin Pinkney wrote one that works for me in 1024 and 512: https://github.com/dvschultz/stylegan2-ada-pytorch/blob/main/export_weights.py

I needed to additionally change generate.py as described here, otherwise this works perfectly