Skip to content
/ DCGAN Public

In this repository, I reproduce the DCGAN paper using PyTorch

License

Notifications You must be signed in to change notification settings

iArunava/DCGAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

87 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

  1. Go to the Colab Notebook: DCGAN Colab

  1. 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
  1. 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

fake_41_faces

  1. Stanford Cars Dataset

Epoch 100

  1. Oxford Flowers Dataset

Epoch 200

  1. Simpsons Faces

fake_99 (2)

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.

About

In this repository, I reproduce the DCGAN paper using PyTorch

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published