Implementation of Convolutional DRAW on MNIST, SVHN and CIFAR-10 in TensorFlow.
Here are some examples of reconstructions produced by the model:
CIFAR-10 | SVHN |
---|---|
You just need to run the python script for a given dataset. For example, python draw-svhn.py
downloads the svhn
dataset and trains the convolutional DRAW model. After training, output data is written to /tmp/draw/draw_data.npy
; during training we generate reconstructions every 100 iterations.
This code is based on the implementation of vanilla DRAW by Eric Jang.
Send feedback to Volodymyr Kuleshov.