richzhang/colorization

Health v2 · maintenance only

Flags: no_releases

Adoption not part of the score

3461 stars · 922 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-29, confidence not recorded

A Python library implementing automatic colorization of grayscale photos using deep neural networks from the ECCV 2016 'Colorful Image Colorization' paper and the SIGGRAPH 2017 follow-up. It provides pretrained PyTorch models (eccv16 and siggraph17) plus a demo script for colorizing images.

Use cases

• colorize black and white photos automatically
• add plausible colors to grayscale images with a pretrained model
• load a colorization model in Python for image restoration
• restore old family photographs with deep learning
• compare ECCV16 and SIGGRAPH17 colorization models
• batch colorize historical photo archives

When to choose

• you need automatic, no-user-input colorization of grayscale images
• you want a simple pretrained PyTorch model for image colorization
• you want to reproduce results from the Colorful Image Colorization paper

When to avoid

• you need interactive or user-guided colorization with fine control (see the authors' interactive deep colorization work)
• you need training code or modern maintained tooling - the original Caffe branch is unsupported
• you require guaranteed accurate colors rather than plausible hallucinated ones

Facets

library · maturity maintenance

image-processing computer-vision deep-learning computer-vision image-processing deep-learning artificial-intelligence python cross-platform colorization grayscale-images pytorch pretrained-models eccv-2016 caffe

Member repositories

For agents

markdown · JSON · MCP: product_card(name="richzhang/colorization")