Image quantization using machine learning

Abstract

Methods and systems are disclosed for quantizing images using machine-learning. A plurality of input images are received from a sensor (e.g., a camera), wherein each input image includes a plurality of pixels. Utilizing an image-to-image machine-learning model, each pixel is assigned a new pixel color. Utilizing a mixer machine-learning model, each new pixel color is converted to one of a fixed number of colors to produce a plurality of quantized images, with each quantized image corresponding to one of the input images. A loss function is determined based on an alignment of each input image with its corresponding quantized image via a pre-trained reference machine-learning model. One or more parameters of the image-to-image machine-learning model and the mixer model are updated based on the loss function. The process repeats, with each iteration updating the parameters of the image-to-image machine-learning model and the mixer model, until convergence, resulting in trained models.