sorin-postelnicu/dcgan

In this repository, I reproduce the DCGAN paper using PyTorch

DCGAN

In this repository, I have reproduced the DCGAN paper. The paper can be found here: DCGAN The implementation is done using PyTorch. I have also released pretrained models. Also, I am publishing a colab notebook, with which you can reproduce the entire model in just one go. And with the help of the pretrained models, you can also, see amazing results, right in your browser and start playing with it.

How to use

0. Go to the Colab Notebook: DCGAN Colab

---

0. Clone the repository and cd into it

git clone https://github.com/iArunava/DCGAN.git
cd DCGAN/

1. Start training the model

# To train the model of the Oxford Flowers Dataset
python3 init.py --mode train -dt flowers


# To train the model of the Stanford Cars Dataset
python3 init.py --mode train -dt cars


# To train the model of the Dogs Dataset
python3 init.py --mode train -dt dogs

2. Test the model using pretrained model

python3 init.py --mode predict -gpath /path/to/pretrained/generator/model.pth

Note: You can download the pretrained model, the links are available in the readme of the flowers, cars directories under results directories.

Samples generated using DCGAN

1. Faces Dataset

2. Stanford Cars Dataset

3. Oxford Flowers Dataset

4. Simpsons Faces

The images were resized to 64x64 before training

Pretrained models

1. Generator trained on the Stanford Cars Dataset for 100 epochs: 100 epoch trained Generator

2. Generator trained on the Oxford Flowers Dataset for 300 epochs: 300 epoch trained Generator

3. Generator trained on the Simpsons Faces Dataset for 300 epochs: 300 epoch trained Generator

License

The code in this repository is made available for free. Feel free to fork and go crazy. Also, you are welcome to use this code in commercial purposes for free, with just proper linkage redirecting back to this repository.