yunjey / domain-transfer-network

TensorFlow Implementation of Unsupervised Cross-Domain Image Generation
MIT License
862 stars 203 forks source link
domain-transfer image-generation tensorflow unsupervised-learning

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