/domain-transfer-network

TensorFlow Implementation of "Unsupervised Cross-Domain Image Generation"

Primary LanguagePythonMIT LicenseMIT

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

From SVHN to MNIST

alt text

alt text

alt text

alt text

From Photos to Emoji (in paper)

alt text

alt text