Name Last Update
jpg Loading commit data...
DTN.pdf Loading commit data...
LICENSE Loading commit data...
README.md Loading commit data...
config.py Loading commit data...
download.sh Loading commit data...
main.py Loading commit data...
model.py Loading commit data...
ops.py Loading commit data...
solver.py Loading commit data...

Domain Transfer Network (DTN)

TensorFlow implementation of Unsupervised Cross-Domain Image Generation.

alt text

Requirements

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 f

python main.py --mode='pretrain'

Train the model G and D

python main.py --mode='train'

Transfer SVHN to MNIST

python main.py --mode='eval'

Results

alt text