yunjey / mnist-svhn-transfer

PyTorch Implementation of CycleGAN and SSGAN for Domain Transfer (Minimal)

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