Skip to content

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

License

Notifications You must be signed in to change notification settings

Notlaughinggirl/mnist-svhn-transfer

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

30 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MNIST-to-SVHN and SVHN-to-MNIST

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

alt text

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 alt text
alt text alt text

2) SGAN

From SVHN to MNIST From MNIST to SVHN
alt text alt text
alt text alt text

About

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

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 98.3%
  • Shell 1.7%