Skip to content

Latest commit

 

History

History
18 lines (12 loc) · 1.06 KB

README.md

File metadata and controls

18 lines (12 loc) · 1.06 KB

CAT GENERATOR

A Deep Convolutional GAN [original paper] trained on Cat-faces-dataset. The dataset has 30k pictures of cats of size 64x64, and each image contains only the face of the cats.

The GAN was implemented in PyTorch from DCGAN TUTORIAL.

  • cat_generator.ipynb GAN training and results visualization.
  • mygan.py custom classes.

The DCGAN converges to realistic pictures with only 50 epochs (already overkilling without at least one or two GPUs), but some more hyperparameters tuning (and more epochs!) could lead to better results. Despite there being a mode collapse in some of the images generated, I ended up with some really nice fake cats.