Skip to content

Train an autoencoder to encode partially drawn images from Google's Quickdraw Dataset. This autoencoder can then be used to bootstrap downstream tasks such as image classification or stroke generation.

Notifications You must be signed in to change notification settings

kylesayrs/SketchCNN-AE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

37 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SketchCNN AutoEncoder

Train an autoencoder to encode partially drawn images from Google's Quickdraw Dataset. This autoencoder can then be used to bootstrap downstream tasks such as image classification or stroke generation.

Example autoencoder output

Usage

python3 download_bitmaps.py drawings
python3 train.py

TODO

Implement data augmentations for strokes

About

Train an autoencoder to encode partially drawn images from Google's Quickdraw Dataset. This autoencoder can then be used to bootstrap downstream tasks such as image classification or stroke generation.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published