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'

Results