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 |
 |
 |
Training on 256x256
- batch size 32
- around 300k steps
| d_loss |
g_loss |
 |
 |
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