Convert StyleGAN2 TensorFlow weights to Pytorch

How to generate a realistic image with Pytorch’s StyleGAN

I want to play with StyleGAN

The official repository of StyleGAN2 is TensorFlow1.
I will convert it to Pytorch and use it.

stylegan2-pytorch

Clone the official repository and the pytorch version of the repository.

GitHub - NVlabs/stylegan2: StyleGAN2 - Official TensorFlow Implementation

GitHub - rosinality/stylegan2-pytorch: Implementation of Analyzing and Improving the Image Quality…

Convert the original checkpoint file pkl to pytorch format.
There are various pre-trained weights such as anime.

GitHub - justinpinkney/awesome-pretrained-stylegan2: A collection of pre-trained StyleGAN 2 models…

python convert_weight.py --repo ../stylegan2 {path_to_checkpoint.pkl}

The Pytorch weight pt file and the original and pytorch version image generation comparison png files are saved.

The original and converted versions have the same output.

Generate an image with the converted weight pt.

python generate.py --sample 1 --pics 1 --ckpt {converted_checkpoint_path.pt}

🐣

I’m a freelance engineer.
Work consultation
Please feel free to contact us with a brief development description.
rockyshikoku@gmail.com

I am making an app that uses Core ML and ARKit.
We send machine learning / AR related information.

GitHub

Twitter
Medium