Section: New Software and Platforms
QuantizationAE
Keywords: Compression - Machine learning
Functional Description: This code learns an autoencoder to compress images. The learning is performed under a rate-distortion criterion, and jointly learns a transform (the autoencoder) and the quantization step for target rate points.The code is organized as follows. It first builds a set of luminance images (B1) for the auto-encoder training, a set of luminance images (B2) to analyze how the auto-encoder training advances and a set of luminance images (B3) to evaluate the auto-encoders in terms of rate-distortion. It then trains several auto-encoders using a rate-distortion criterion on the set B1. The quantization can be either fixed or learned during this training stage. The set B2 enables to periodically compute indicators to detect overfitting. It finally compares the auto-encoders in terms of rate-distortion on the set B3. The quantization can be either fixed or variable during this test.