a DRAW model for colored images in tensorflow
Latest commit cf971ad Dec 5, 2016 @kvfrans readup
Failed to load latest commit information.
good-results plotted dots Dec 4, 2016
results-noattn plotted dots Dec 4, 2016
results blank results Dec 4, 2016
.DS_Store rm junk Oct 27, 2016
README.md readup Dec 6, 2016
color.py plotted dots Dec 4, 2016
input_data.py init Sep 5, 2016
main.py color and results Oct 30, 2016
ops.py init Sep 5, 2016
plot_data.py color working Oct 27, 2016
res_attention.gif colors Oct 27, 2016
res_noattn.gif colors Oct 27, 2016
utils.py plotted dots Dec 4, 2016



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.


python main.py

The colored DRAW runs on the celebA dataset.


python color.py