autocolorize 0.2.1

Automatic colorizaton of grayscale images using Deep Learning.

autocolorize

Automatically colorize images using Machine Learning.

• Project page

• arXiv paper

Installation

Make sure that you have Caffe (with Python bindings). Then run:

pip install autocolorize

Run:

autocolorize grayscale.png -o colorized.png

API

You can also colorize from Python (assuming grayscale is the image that you want to colorize):

import autocolorize
classifier = autocolorize.load_default_classifier()
rgb = autocolorize.colorize(grayscale, classifier=classifier)