Patent ID: 11887219
Assignee: GOOGLE LLC
Field: Computer technology (Electrical engineering)
Classification: CPC G | IPC G  H

Claim 12:
13. A computing system comprising:
one or more processors;
a colorization model configured to receive a training input image and, in response to receiving the training input image, output a predicted color map that describes a predicted colorization for the training input image;
one or more non-transitory computer-readable media that collectively store instructions that, when executed by the one or more processors, cause the computing system to perform operations, the operations comprising:
inputting the training input image into the colorization model;
receiving the predicted color map as an output of the colorization model;
calculating a first color distance between a first pixel included in the predicted color map and a second pixel included in the predicted color map;
calculating a second color distance between a third pixel included in a ground truth color map and a fourth pixel included in the ground truth color map, wherein the third pixel and fourth pixels included in the ground truth color map spatially correspond, respectively, with the first pixel and second pixel included in the predicted color map;
evaluating a neighborhood color loss function that evaluates a difference between the first color distance and the second color distance; and
adjusting parameters associated with the colorization model based on the neighborhood color loss function; and
iteratively generating respective neighborhood color loss maps for respective additional first pixels of the predicted color map.