Skip to content
master
Switch branches/tags
Code

Latest commit

Removes dependencies matio and gason. They can be compiled in by
specifiying USE_MATIO and USE_GASON, respectively.

Fixes #6
Fixes #7
79d4ce1

Git stats

Files

Permalink
Failed to load latest commit information.

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.

About

Automatic colorizaton of grayscale images using Deep Learning

Resources

License

Packages

No packages published