DCGAN
Explore the docs »

Report a Bug · Request a Feature . Ask a Question

Table of Contents

• About
• Prerequisites
• Usage
• License
• Acknowledgements

---

About

This repo is the unofficial pytorch implementation of MNIST and CelebA image generating using Deep Convolutional Generative Adversarial Network (DCGAN)

Prerequisites

see requirements.txt

Usage

1. Clone the repo
2. Install the requirements
3. If training on CelebA dataset, add kaggle.json to the root directory by following the instructions here
4. Run the following command to train the model

python ./train.py --dataset [MNIST/CelebA]

License

This project is licensed under the MIT license.

See LICENSE for more information.

Acknowledgements

https://github.com/aladdinpersson/Machine-Learning-Collection/tree/master/ML/Pytorch/GANs/