Automatic colorizaton of grayscale images using Deep Learning
Switch branches/tags
Clone or download
gustavla Fixes ExtendedImageDataLayer
Removes dependencies matio and gason. They can be compiled in by
specifiying USE_MATIO and USE_GASON, respectively.

Fixes #6
Fixes #7
Latest commit 79d4ce1 Sep 5, 2017

README.rst

autocolorize

Automatically colorize images using Machine Learning.

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)

Sparse training

We provide custom layers for doing sparse hypercolumn training in both Caffe (see caffe/) and Tensorflow (see tensorflow/). This can be used for other image-to-image tasks, such as semantic segmentation or edge prediction.

Look inside the train folder if you want to train from scratch.