Skip to content

tymokvo/AEGeAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AEGeAN

Deeper DCGAN with AE stabilization

Parallel training of generative adversarial network as an autoencoder with dedicated losses for each stage. Generator class has conditional .forward() method for enhanced ergonomics. Autoencoding pass seems to avoid mode collapse and recover faster if Generator is not doing well.

Has been used successfully with as few as ~200 images in the source folder.

Builds on the DCGAN PyTorch demo. This one generates images upto 1024x1024 so it can use a lot of VRAM.

Should work when the "dataroot" is configured ImageNet style: ".../a_dir_of_images/what_would_be_a_label" or ".../cat_pics/cute_cats/cat_001.jpg"

Have fun!

Examples of generated drawings here

A description of the project for which this was developed can be found here.

About

Deeper DCGAN with AE stabilization

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages