Home

Awesome

MNIST-to-SVHN and SVHN-to-MNIST

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

alt text

Prerequites

<br>

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
<br>

Results

1) CycleGAN (should be re-uploaded)

From SVHN to MNISTFrom MNIST to SVHN
alt textalt text
alt textalt text

2) SGAN

From SVHN to MNISTFrom MNIST to SVHN
alt textalt text
alt textalt text