KulkarniKaustubh/PokeGAN

A DCGAN to generate Pokemon

Pokemon Generator

A DCGAN to generate Pokemon. The dataset I have used has ~2800 images shrunk down to (100, 100) from the original size, and each model has been trained over 50 epochs.

(I have used a smaller dataset as GitHub has a file size limit of 100MB)

The original dataset that consists of ~11,000 images can be found here

Images

Here are a few images generated across a couple of models.

Models

1. • Learning rate 1e-4
   • Adam optimizer for both generator and discriminator

   

2. • Learning rate 1e-4
   • Adam optimizer for generator and SGD optimizer for discriminator

   

3. • Learning rate 1e-4
   • Adam optimizer with momentum = 0.5 for generator
   • SGD optimizer for discriminator

   

Images

All images shown above can be found here

Graphs

The graphs for the different models over 50 epochs can be found here

Code

Code is in the src folder