BEGAN in Tensorflow

Requirements

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

Name		Name	Last commit message	Last commit date
Latest commit History 37 Commits
assets		assets
.gitignore		.gitignore
README.md		README.md
config.py		config.py
data_loader.py		data_loader.py
download.py		download.py
folder.py		folder.py
layers.py		layers.py
main.py		main.py
models.py		models.py
trainer.py		trainer.py
utils.py		utils.py