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