qbxlvnf11 / conditional-GAN

Pytorch implementation of Conditional-GAN (CGAN)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Description

- Conditional GAN (CGAN)

  • Conditional version of generative adversarial nets
    • In an unconditioned generative model, there is no control on modes of the data being generated.
    • In the Conditional GAN (CGAN), the generator learns to generate a fake sample with a specific condition or characteristics (such as a label associated with an image or more detailed tag) rather than a generic sample from unknown noise distribution.
    • Simply feeding the data y, and conditioning both the generator and discriminator
  • Adversarial learning loss of CGAN

Contents

- Generating Fashion MNIST Image with CGAN

  • Generator
    • Input: z (100 dimension), Output: generated image
    • FC [256, 512, 1024]
  • Discriminator
    • Input: image (generated or real), Output: fake/real
    • FC [1024, 512, 256]
  • Adversarial learning: refer to generator_train_step(), discriminator_train_step()

Dataset

- Fashion MNIST

https://www.kaggle.com/datasets/zalando-research/fashionmnist

References

- CGAN code

https://www.kaggle.com/code/arturlacerda/pytorch-conditional-gan/notebook

- CGAN Paper

@article{CGAN,
  title={Conditional Generative Adversarial Nets},
  author={Mehdi Mirza, Simon Osindero},
  journal = {arXiv},
  year={2014}
}

Author

- LinkedIn: https://www.linkedin.com/in/taeyong-kong-016bb2154

- Blog URL: https://blog.naver.com/qbxlvnf11

- Email: qbxlvnf11@google.com, qbxlvnf11@naver.com

About

Pytorch implementation of Conditional-GAN (CGAN)

License:MIT License


Languages

Language:Jupyter Notebook 100.0%