Automatic colorizaton of grayscale images using Deep Learning
Switch branches/tags
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
autocolorize
caffe
scripts
tensorflow
train
.gitignore
LICENSE
MANIFEST.in
README.rst
requirements.txt
requirements_docs.txt
setup.cfg
setup.py

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.