According to one implementation, an image compression system includes a computing platform having a hardware processor and a system memory storing a software code. The hardware processor executes the software code to receive an input image, transform the input image to a latent space representation of the input image, and quantize the latent space representation of the input image to produce multiple quantized latents. The hardware processor further executes the software code to encode the quantized latents using a probability density function of the latent space representation of the input image, to generate a bitstream, and convert the bitstream into an output image corresponding to the input image. The probability density function of the latent space representation of the input image is obtained based on a normalizing flow mapping of one of the input image or the latent space representation of the input image.
BACKGROUND (1) The field of neural image compression has made significant progress with deep learning based approaches. There are two key aspects in neural based image compression. First a mapping from image space to a latent space must be learned. Second, a probability needs to be learned in this new space to allow entropy coding of the latents. (2) Conventional approaches have addressed the first aspect by proposing neural network architectures to parameterize the encoding and decoding functions typically needed to achieve good compression results. More recently, the primary focus of research has been on the second aspect: trying to accurately model the distribution in the latent space, where there remains a need in the art for additional progress. SUMMARY (3) There are provided systems and methods for performing image compression using normalizing flows, substantially as shown in and/or described in connection with at least one of the figures, and as set forth more completely in the claims.
1. An image compression system comprising: a computing platform including a hardware processor and a system memory; a software code including a normalizing flow stored in the system memory; the hardware processor configured to execute the software code to: receive an input image by the normalizing flow; transform, by the normalizing flow serving as an alternative to an encoder, the input image to a latent space representation of the input image; quantize the latent space representation of the input image to produce a plurality of quantized latents; obtain a probability density function of the latent space representation of the input image based on a normalizing flow mapping of the latent space representation of the input image by the normalizing flow; generate a bitstream using the plurality of quantized latents and the probability density function of the latent space representation of the input image; and convert the bitstream into an output image corresponding to the input image. ||
7. A method for use by an image compression system including a computing platform having a hardware processor and a system memory storing a software code including a normalizing flow, the method comprising: receiving, by the normalizing flow of the software code, executed by the hardware processor, an input image; transforming, by the software code executed by the hardware processor, the input image to a latent space representation of the input image using the normalizing flow serving as an alternative to an encoder; quantizing, by the software code executed by the hardware processor, the latent space representation of the input image to produce a plurality of quantized latents; obtaining, by the software code executed by the hardware processor, a probability density function of the latent space representation of the input image based on a normalizing flow mapping of the latent space representation of the input image by the normalizing flow; generating a bitstream, by the software code executed by the hardware processor, using the plurality of quantized latents and the probability density function of the latent space representation of the input image; and converting, by the software code executed by the hardware processor, the bitstream into an output image corresponding to the input image. ||
12. An image compression system comprising: a computing platform including a hardware processor and a system memory; a software code stored in the system memory; the hardware processor configured to execute the software code to: receive an input image; transform the input image to a latent space representation of the input image using a first normalizing flow; quantize the latent space representation of the input image to produce a plurality of quantized latents; obtain a probability density function of the latent space representation of the input image based on a first normalizing flow mapping of the latent space representation of the input image by the first normalizing flow; encode the plurality of quantized latents using the probability density function of the latent space representation of the input image, to generate a bitstream; and convert the bitstream into an output image corresponding to the input image using a second normalizing flow.