kylesayrs / SketchCNN-AE

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.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.


Languages

Language:Python 97.8%Language:Shell 2.2%