Skip to content

ssidorenko/clouds_dcgan

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Generating clouds using a 128x128 DCGAN

This code is based on the original pytorch DCGAN example, modified to output 128x128 images instead of 64x64 in the original example.

From 64x64 to 128x128

A convolution layer has been added to the generator network and to the discriminator network so that they respectively output and discriminate on 128x128 images. To stabilize training, at each epoch gaussian noise is added to the current training data batch. The number of features of the generator has been increased from 64 to 256 on the generator, but kept to 64 on the discriminator. This seemed to avoid falling into a collapsed mode.

Batch size tuning

The parameters have been adapted to run on a GeForce 1060 6G. Batch size can be increased if more memory is available, but this sometimes bring instability to the training process. Learning rate should be decreased if this happens.

Dataset

Images are from ImageNet's cloud synset.

Examples

Images generated by randomly sampling the representation space after training the networks for 700 epochs: sample generated images

Fixed random vector on the representation space while training: sample generated images

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published