Image colorization using machine learning

Abstract

Implementations described herein relate to methods, systems, and computer-readable media to train and use a machine-learning model to colorize a grayscale image that depicts a person. In some implementations, a computer-implemented method includes receiving the grayscale image. The method further includes generating a colorized image based on the grayscale image as output of a trained convolutional neural network (CNN) by providing the grayscale image as input to the trained CNN. In some implementations, the trained CNN performs part segmentation to detect one or more parts of the person and colorizes the grayscale image.