Skip to content

gsurma/image_generator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image Generator

Python notebook containing TensorFlow DCGAN implementation. It was trained on a Simpsons Faces dataset.


Check out corresponding Kaggle kernel: Image Generator.

Check out corresponding Medium article:

Image Generator - Drawing Cartoons with Generative Adversarial Networks

DCGAN

Network architecture by Radford et al., 2015.

Training

Visualization of training with the following hyperparameteres.

IMAGE_SIZE = 128
NOISE_SIZE = 100
LR_D = 0.00004
LR_G = 0.0004
BATCH_SIZE = 64
EPOCHS = 300
BETA1 = 0.5
WEIGHT_INIT_STDDEV = 0.02
EPSILON = 0.00005

Results

Cherry-picked generated samples.

As expected, there were some funny-looking malformed faces as well.

Author

Greg (Grzegorz) Surma

PORTFOLIO

GITHUB

BLOG