Colorization by Matrix Completion

Authors

  • Shusen Wang Zhejiang University
  • Zhihua Zhang Zhejiang University

DOI:

https://doi.org/10.1609/aaai.v26i1.8274

Keywords:

colorization, matrix recovery, matrix completion, convex optimization

Abstract

Given a monochrome image and some manually labeled pixels, the colorization problem is a computer-assisted process of adding color to the monochrome image. This paper proposes a novel approach to the colorization problem by formulating it as a matrix completion problem. In particular, taking a monochrome image and parts of the color pixels (labels) as inputs, we develop a robust colorization model and resort to an augmented Lagrange multiplier algorithm for solving the model. Our approach is based on the fact that a matrix can be represented as a low-rank matrix plus a sparse matrix. Our approach is effective because it is able to handle the potential noises in the monochrome image and outliers in the labels. To improve the performance of our method, we further incorporate a so-called local-color-consistency idea into our method. Empirical results on real data sets are encouraging.

Downloads

Published

2021-09-20

How to Cite

Wang, S., & Zhang, Z. (2021). Colorization by Matrix Completion. Proceedings of the AAAI Conference on Artificial Intelligence, 26(1), 1169-1175. https://doi.org/10.1609/aaai.v26i1.8274

Issue

Section

AAAI Technical Track: Machine Learning