Skip to content

gcroci2/CatGenerator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published