SketchRNN Implementation In Tensorflow 2

This repo contains a simple implementation of the SketchRNN model with Tensorflow 2, following the best practice as much as possible.

Setup

After cloning the repo, run the following commands to setup

pipenv install --dev --skip-lock
pipenv shell

See the colab notebook for an example train on the rabbit dataset.

Here are some results from a model trained on the rabbit dataset.

See complete examples in the usage.ipynb.

A Neural Representation of Sketch Drawings - Original Paper
Teaching Machines to Draw - Official blog post
googlecreativelab/quickdraw-dataset - The QuickDraw dataset
magenta/models/sketch_rnn - Official Implementation
hardmaru/sketch-rnn - Original Implementation from the author of the paper @hardmaru
eyalzk/sketch_rnn_keras - Implementation based on Keras by @eyalzk
zzh8829/yolov3-tf2 - Great source for some TF2 features in practice.