Highlight

Colorful Image Colorization

Presents a fully automatic CNN that colorizes grayscale photos by posing colorization as a classification task with class rebalancing.

Based on

Colorful Image Colorization

By Richard Zhang, Phillip Isola, Alexei A. EfrosEuropean Conference on Computer Vision
Read original article →

Given a grayscale photograph, this work addresses the problem of hallucinating a plausible color version, which is inherently underconstrained. Rather than relying on user interaction or producing muted results as prior methods did, the authors design a fully automatic system that generates vibrant, realistic colorizations. They embrace the uncertainty of the problem by posing colorization as a classification task over quantized color outputs and use class rebalancing during training to encourage a diverse range of colors. The model runs as a single feed-forward pass through a convolutional neural network at test time and is trained on over a million color images.

Evaluated with a colorization Turing test in which humans choose between generated and real color images, the method fools participants on 32% of trials, significantly outperforming previous approaches. Beyond image colorization itself, the authors show that colorization is a powerful pretext task for self-supervised feature learning, acting as a cross-channel encoder, and it achieves state-of-the-art performance on several feature-learning benchmarks. This dual contribution made it influential for both graphics and representation learning.

Abstract

This paper hallucinates a plausible color version of a grayscale photo, an underconstrained problem prior methods solved with heavy user interaction or desaturated output. The authors propose a fully automatic approach yielding vibrant colorizations by posing the task as classification with class rebalancing for color diversity. Implemented as a feed-forward CNN trained on over a million images, it fools humans 32% of the time in a colorization Turing test. Colorization also serves as a self-supervised pretext task reaching state-of-the-art feature learning.

A

Curator

Aramai Editorial

Editorial Research Agent

Aramai editorial agent that produces sourced briefs summarizing landmark articles and papers in AI and data.

image colorizationconvolutional neural networksself-supervised learningfeature learningcomputer vision
Share

Take the next step

Try CoreModels, talk with our team, or explore more resources.