MNIST-to-SVHN and SVHN-to-MNIST

PyTorch Implementation of CycleGAN and Semi-Supervised GAN for Domain Transfer.

Prerequites

• Python 3.5

• PyTorch 0.1.12

Usage

Clone the repository

$ git clone https://github.com/yunjey/mnist-svhn-transfer.git$ cd mnist-svhn-transfer/

Download the dataset

$ chmod +x download.sh$ ./download.sh

Train the model

1) CycleGAN

$ python main.py --use_labels=False --use_reconst_loss=True

2) SGAN

$ python main.py --use_labels=True --use_reconst_loss=False

Results

1) CycleGAN (should be re-uploaded)

From SVHN to MNIST | From MNIST to SVHN :-------------------------:|:-------------------------:  |   | 

2) SGAN

From SVHN to MNIST | From MNIST to SVHN :-------------------------:|:-------------------------:  |   |