| # 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](https://github.com/tkarras/progressive_growing_of_gans) (TensorFlow) | |
| - [StyleGAN](https://github.com/NVlabs/stylegan) (TensorFlow) | |
| - [StyleGAN2](https://github.com/NVlabs/stylegan2) (TensorFlow) | |
| - [StyleGAN2-ADA](https://github.com/NVlabs/stylegan2-ada) (TensorFlow) | |
| - [StyleGAN2-ADA-PyTorch](https://github.com/NVlabs/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](https://github.com/tkarras/progressive_growing_of_gans) | |
| - `stylegan_official/`: [StyleGAN](https://github.com/NVlabs/stylegan) | |
| - `stylegan2_official/`: [StyleGAN2](https://github.com/NVlabs/stylegan2) | |
| - `stylegan2ada_tf_official/`: [StyleGAN2-ADA](https://github.com/NVlabs/stylegan2-ada) | |
| - `stylegan2ada_pth_official/`: [StyleGAN2-ADA-PyTorch](https://github.com/NVlabs/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 | |
| ```shell | |
| 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. | |