This repo contains demo code to run the colorization model described in:

Colorful Image Colorization
Richard Zhang, Phillip Isola, Alexei A. Efros
In arXiv, 2016
Project Page

We include demo usage as an iPython notebook, under ./demo/colorization_demo_v0.ipynb. Clone the master branch of the respository using git clone -b master --single-branch https://github.com/richzhang/colorization.git, cd into the demo directory, run ipython notebook and open colorization_demo_v0.ipynb in your web browser.

This demo code requires a working installation of Caffe and basic Python libraries (numpy, pyplot, skimage, scipy). For guidelines and help with installation of Caffe, consult the installation guide, Caffe users group, and Caffe Github site.

Please contact Richard Zhang at [email protected] for any questions or comments. Should you find this model useful for your resesarch, please use this bibtex to cite.