Augmenting training data by recoloring images

Abstract

Methods, systems, and apparatus, including computer programs encoded on computer storage media, for generating, using one or more coloraF_ization models, an augmented set of training data. One of the methods includes receiving a plurality of training examples for training an image processing model, each training example comprising an image and a corresponding ground-truth output for the image; generating, for each image, a respective grayscale image; generating, for each respective grayscale image, one or more recolored images using one or more colorization models; and generating an augmented set of training data for training the image processing model that comprises a plurality of additional training examples.