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.
- Go to the Colab Notebook: DCGAN Colab
- Clone the repository and cd into it
git clone https://github.com/iArunava/DCGAN.git
cd DCGAN/
- 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
- 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.
- Faces Dataset
- Stanford Cars Dataset
- Oxford Flowers Dataset
.png)
- Simpsons Faces
The images were resized to 64x64 before training
-
Generator trained on the Stanford Cars Dataset for 100 epochs: 100 epoch trained Generator
-
Generator trained on the Oxford Flowers Dataset for 300 epochs: 300 epoch trained Generator
-
Generator trained on the Simpsons Faces Dataset for 300 epochs: 300 epoch trained Generator
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.