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

Prerequisite

• pyenv

• pipenv

Installation

After cloning the repo, run the following commands to setup

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

Usage

Training

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

Results

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

References

• 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.