Tensorflow Framework

A framework for building, testing, and training Tensorflow networks.

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.)

Dataset setup:

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