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

Claim 4:
5. The method of claim 1, wherein:
the colorization model comprises a multinomial prediction model and a refinement model, wherein the multinomial prediction model is configured to receive the training input image, and in response to receiving the training input image, output a multinomial color distribution that describes a plurality of colorizations for the training input image in a discretized color space, and wherein the refinement model is configured to receive the multinomial color distribution, and in response to receiving the multinomial color distribution, output the predicted color map;
inputting the training input image into the colorization model comprises inputting the training input image into the multinomial prediction model; and
the method further comprises:
receiving the multinomial color distribution as an output of the multinomial prediction model; and
inputting the multinomial color distribution into the refinement model.