Conditional-GANs-Pytorch
Conditional-GANs-Pytorch copied to clipboard
CGAN ProjectionCGAN ACGAN InfoGAN Pytorch
Conditional GANs
Pytorch implementation of several GANs with conditional signals (supervised or unsupervised). All experiments are conducted on Fashion-MNIST, and the network structures are adapted from Improved GAN.
Conditional GANs
- Supervised
- [x] CGAN
- [x] CGAN with Projection Discriminator
- [x] ACGAN
- [ ] Others
- Unsupervised
- [x] InfoGAN
- [ ] Others
Exemplar Results
| CGAN | Projection CGAN | ACGAN |
|---|---|---|
![]() |
![]() |
![]() |
| InfoGAN1 | InfoGAN2 | InfoGAN3 |
![]() |
![]() |
![]() |
Usage
-
Prerequisites
- PyTorch 1.0.0
- Python 3.6
-
Examples of training
-
training
CUDA_VISIBLE_DEVICES=0 python train_CGAN.py -
tensorboard for loss visualization
CUDA_VISIBLE_DEVICES='' tensorboard --logdir ./output/CGAN_default/summaries --port 6006
-
-
Others
- If you want to use other datasets, just replace
FashionMNISTbyMNISTorCIFAR10in the codes. - There are arguments for configurations of GAN loss, gradient penalty, and etc, just try them.
- If you want to use other datasets, just replace





