Tensorflow implementation of BEGAN: Boundary Equilibrium Generative Adversarial Networks.
- Python 2.7
- Pillow
- tqdm
- requests (Only used for downloading CelebA dataset)
- TensorFlow 1.1.0 (Need nightly build which can be found in here, if not you'll see
ValueError: 'image' must be three-dimensional.)
First download CelebA datasets with:
$ apt-get install p7zip-full # ubuntu
$ brew install p7zip # Mac
$ python download.py
or you can use your own dataset by placing images like:
data
└── YOUR_DATASET_NAME
├── xxx.jpg (name doesn't matter)
├── yyy.jpg
└── ...
To train a model:
$ python main.py --dataset=CelebA --use_gpu=True
$ python main.py --dataset=YOUR_DATASET_NAME --use_gpu=True
To test a model (use your load_path):
$ python main.py --dataset=CelebA --load_path=CelebA_0405_124806 --use_gpu=True --is_train=False --split valid
Taehoon Kim / @carpedm20