- This is tensorflow 2.0 based keras subclassing reimplementation of official StyleGAN2 Repo
- Please check how_to_extract_official_weights.md
Official repo result | Official weight copied result from this repo |
---|---|
- batch size 32
- around 300k steps
d_loss | g_loss |
---|---|
- result samples
- Create dataset(tf.data.Dataset) loader which returns
- shape:
[None, 3, res, res]
whereres
is desired image resolution - image value range:
[-1, +1]
- shape:
- 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