a DRAW model for colored images in tensorflow
Python
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
good-results
results-noattn
results
.DS_Store
README.md
color.py
input_data.py
main.py
ops.py
plot_data.py
res_attention.gif
res_noattn.gif
utils.py

README.md

draw-color

A Tensorflow implementation of DRAW. Now includes support for colored images!

This is code to accompany my post on the DRAW model.

For an explanation of how I modified the original DRAW into a colored model, check out my post on colorizing DRAW.

The original DRAW is a slightly rewritten, commented version of ericjang's implementation, running on MNIST.

Usage:

python main.py

The colored DRAW runs on the celebA dataset.

Usage:

python color.py