taki0112 / Stylegan Tensorflow
Licence: mit
Simple & Intuitive Tensorflow implementation of StyleGAN (CVPR 2019 Oral)
Stars: ✭ 198
Programming Languages
python
139335 projects - #7 most used programming language
StyleGAN-Tensorflow
Simple & Intuitive Tensorflow implementation of "A Style-Based Generator Architecture for Generative Adversarial Networks" (CVPR 2019 Oral)
Official code | Paper | Video | FFHQ Dataset
Other implementation
TF Estimator version
Pretrained model
Checkpoint
Usage
├── dataset
└── YOUR_DATASET_NAME
├── 000001.jpg
├── 000002.png
└── ...
Train
> python main.py --dataset FFHQ --img_size 1024 --gpu_num 4 --progressive True --phase train
Test
> python main.py --dataset FFHQ --img_size 1024 --progressive True --batch_size 16 --phase test
Draw
Figure02 uncurated
python main.py --dataset FFHQ --img_size 1024 --progressive True --phase draw --draw uncurated
Figure03 style mixing
python main.py --dataset FFHQ --img_size 1024 --progressive True --phase draw --draw style_mix
Figure08 truncation trick
python main.py --dataset FFHQ --img_size 1024 --progressive True --phase draw --draw truncation_trick
Architecture
Our Results (1024x1024)
- Training time: 2 days 14 hours with V100 * 4
-
max_iteration
= 900-
Official code
= 2500
-
Uncurated
Style mixing
Truncation trick
Generator loss graph
Discriminator loss graph
Author
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].