Model Converters

Introduction

Besides training, we also support converting pre-trained model weights from officially released models and using them for inference. So, if you have already trained some models with the officially open-sourced codebase, don't worry, we have already made sure that they well match our codebase! We now support models trained with following repositories:

PGGAN (TensorFlow)
StyleGAN (TensorFlow)
StyleGAN2 (TensorFlow)
StyleGAN2-ADA (TensorFlow)
StyleGAN2-ADA-PyTorch (PyTorch)

NOTE: Our codebase is completely built on PyTorch. But, if you want to convert the official TensorFlow model, you need to setup the TensorFlow environment. This can be easily done with pip install tensorflow-gpu==1.15.

We also mirror the officially open-sourced codes in this folder, which are relied on by pickle.load(). Specifically, we have

pggan_official/: PGGAN
stylegan_official/: StyleGAN
stylegan2_official/: StyleGAN2
stylegan2ada_tf_official/: StyleGAN2-ADA
stylegan2ada_pth_official/: StyleGAN2-ADA-PyTorch

NOTE: These codes will ONLY be used for model conversion. After that, all codes within this folder will not be used anymore.

Usage

The script to convert a model is provided as ../convert_model.py. For example, to convert a pre-trained StyleGAN2 model (officially TensorFlow version), just run

cd ..
python convert_model.py stylegan2 \
       --source_model_path ${SOURCE_MODEL_PATH} \
       --test_num 10 \
       --save_test_image

The above command will execute the conversion and then test the conversion precision.