MNIST-to-SVHN and SVHN-to-MNIST
PyTorch Implementation of CycleGAN and Semi-Supervised GAN for Domain Transfer.
Prerequites
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 |
|
|
|
|