All Projects → moono → stylegan2-tf-2.x

moono / stylegan2-tf-2.x

Licence: other
stylegan2, tensorflow 2, keras subclassing

Programming Languages

python
139335 projects - #7 most used programming language
Cuda
1817 projects
Dockerfile
14818 projects

stylegan2-tf-2.x

  • This is tensorflow 2.0 based keras subclassing reimplementation of official StyleGAN2 Repo

Inference from official weights

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
# 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
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].