- CelebA
- Official CelebA-HQ
- Unofficial forked CelebA-HQ. I used this 256x256 dataset.
- Download the above dataset and run the below command.
python train.py
- If you have high-end GPUs(e.g. Tesla V100), you can train higher resolution images. I have only RTX 2070....
- To improve the quality, you may change some hyperparmeters(e.g. # of Convolution feature maps (256 -> 512), batch size (>32, important), and update schedule)
- T. Karras' Official Tensorflow implementation