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: SVHN to MNIST Generation

alt text

alt text