MNIST-to-SVHN and SVHN-to-MNIST
PyTorch Implementation of CycleGAN and Semi-Supervised GAN for Domain Transfer.
![alt text](gif/cyclegan.png)
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 |
![alt text](gif/cycle-s-m.gif) |
![alt text](gif/cycle-m-s.gif) |
![alt text](gif/cycle-s-m.png) |
![alt text](gif/cycle-m-s.png) |
2) SGAN
From SVHN to MNIST |
From MNIST to SVHN |
![alt text](gif/sgan-s-m.gif) |
![alt text](gif/sgan-m-s.gif) |
![alt text](gif/sgan-s-m.png) |
![alt text](gif/sgan-m-s.png) |