Domain Transfer Network (DTN)
TensorFlow implementation of Unsupervised Cross-Domain Image Generation.
Requirements
- Python 2.7
- TensorFlow 0.12
Usage
Clone the repository
git clone https://github.com/yunjey/dtn-tensorflow.git
cd dtn-tensorflow
Download the dataset
chmod +x download.sh
./download.sh
Resize MNIST dataset to 32x32
python prepro.py
Pretrain the model
python main.py --mode='pretrain'
Train the model
python main.py --mode='train'
Transfer SVHN to MNIST
python main.py --mode='eval'