diff --git a/README.md b/README.md index 69a7ed8..e69de29 100644 --- a/README.md +++ b/README.md @@ -1,49 +0,0 @@ -# stylegan2-tf-2.x -* This is tensorflow 2.0 based keras subclassing reimplementation of official [StyleGAN2 Repo](https://github.com/NVlabs/stylegan2) - -## Inference from official weights -* Please check [how_to_extract_official_weights.md](./how_to_extract_official_weights.md) - -| Official repo result | Official weight copied result from this repo | -| :---: | :---: | -| ![official_result] | ![restored_result] | - -## Training on 256x256 -* batch size 32 -* around 300k steps - -| d_loss | g_loss | -| :---: | :---: | -| ![256x256_d_loss]| ![256x256_g_loss]| - -* result samples - -| | | | -| :---: | :---: | :---: | -| ![256x256_result_0]| ![256x256_result_1]| ![256x256_result_2]| - -## Using custom data -* Create dataset(tf.data.Dataset) loader which returns - * shape: `[None, 3, res, res]` where `res` is desired image resolution - * image value range: `[-1, +1]` - -## Etc -### Using with pycharm -* If .bashrc file's paths don't work in pycharm environment (e.g. `nvcc -h`), open pycharm with following -```bash -# move to pycharm installed location (location may vary) -moono@moono-ubuntu:~$ cd .local/share/JetBrains/Toolbox/apps/PyCharm-P/ch-0/201.7846.77/bin/ - -# launch pycharm with .bashrc -moono@moono-ubuntu:~/.local/share/JetBrains/Toolbox/apps/PyCharm-P/ch-0/201.7846.77/bin$ bash pycharm.sh -``` - -[loss_tensorboard]: assets/tf-keras-stylegan2-loss.PNG -[generation_tensorboard]: assets/tf-keras-stylegan2-fake-images.PNG -[official_result]: assets/seed6600-official.png -[restored_result]: assets/seed6600-restored.png -[256x256_d_loss]: assets/d_loss_256x256.png -[256x256_g_loss]: assets/g_loss_256x256.png -[256x256_result_0]: assets/out_256x256_0.png -[256x256_result_1]: assets/out_256x256_1.png -[256x256_result_2]: assets/out_256x256_2.png \ No newline at end of file