Tensorflow implementation of BSD-GAN
please cite the paper, if the codes/dataset has been used for your research.
-
Python (2.7 or later)
-
numpy
-
scipy
-
NVIDIA GPU + CUDA 8.0 + CuDNN v5.1
-
TensorFlow 1.0 or later
- clone this repo:
git clone https://github.com/duxingren14/BSD-GAN.git
cd BSD-GAN
mkdir data
- download datasets (e.g., car) from google drive and put it in ./data/.
./BSD-GAN/data/car_400x300/*.jpg
- train the model:
python main.py --phase train --dataset_name car_400x300 --input_height 300 --input_width 400 --epoch 45
- test the model:
python main.py --phase train --dataset_name car_400x300 --input_height 300 --input_width 400 --epoch 45 --train False
some datasets can be downloaded from the website. Please cite their papers if you use the data.
celebahq: https://github.com/tkarras/progressive_growing_of_gans
lsun: http://lsun.cs.princeton.edu/2017/
Codes are built on the top of DCGAN tensorflow and progressive growing. Thanks for their precedent contributions!