Simple working Pytorch implementation of Stylegan2 based on https://arxiv.org/abs/1912.04958
Below are some flowers that do not exist.
Neither do these hands
You will need a machine with a GPU and CUDA installed. Then pip install the package like so
$ pip install stylegan2_pytorch
$ stylegan2_pytorch --data /path/to/images
That's it. Sample images will be saved to results/default
and models will be saved periodically to models/default
.
You can specify the name of your project with
$ stylegan2_pytorch --data /path/to/images --name my-project-name
You can also specify the location where intermediate results and model checkpoints should be stored with
$ stylegan2_pytorch --data /path/to/images --name my-project-name --results_dir /path/to/results/dir --models_dir /path/to/models/dir
By default, if the training gets cut off, it will automatically resume from the last checkpointed file. If you want to restart with new settings, just add a new
flag
$ stylegan2_pytorch --new --data /path/to/images --name my-project-name --image-size 512 --batch-size 1 --gradient-accumulate-every 16 --network-capacity 10
Once you have finished training, you can generate images from your latest checkpoint like so.
$ stylegan2_pytorch --generate
If a previous checkpoint contained a better generator, (which often happens as generators start degrading towards the end of training), you can load from a previous checkpoint with another flag
$ stylegan2_pytorch --generate --load-from {checkpoint number}
- Add mixed precision and multi-GPU support
Thank you to Matthew Mann for his inspiring simple port for Tensorflow 2.0
This also uses the hinge Lโ gradient penalty
described in https://arxiv.org/abs/1910.06922
@article{Karras2019stylegan2,
title = {Analyzing and Improving the Image Quality of {StyleGAN}},
author = {Tero Karras and Samuli Laine and Miika Aittala and Janne Hellsten and Jaakko Lehtinen and Timo Aila},
journal = {CoRR},
volume = {abs/1912.04958},
year = {2019},
}
@article{jolicoeur2019connections}
title={Connections between Support Vector Machines, Wasserstein distance and gradient-penalty GANs},
author={Jolicoeur-Martineau, Alexia},
journal={arXiv preprint arXiv:1910.06922},
year={2019}
}
@article{,
title= {Oxford 102 Flowers},
author= {Nilsback, M-E. and Zisserman, A., 2008},
abstract= {A 102 category dataset consisting of 102 flower categories, commonly occuring in the United Kingdom. Each class consists of 40 to 258 images. The images have large scale, pose and light variations.}
}
@article{afifi201911k,
title={11K Hands: gender recognition and biometric identification using a large dataset of hand images},
author={Afifi, Mahmoud},
journal={Multimedia Tools and Applications},
volume={78},
number={15},
pages={20835--20854},
year={2019},
publisher={Springer}
}