CompressAI: a PyTorch library and evaluation platform for end-to-end compression research
arXiv:2011.03029
Abstract
This paper presents CompressAI, a platform that provides custom operations, layers, models and tools to research, develop and evaluate end-to-end image and video compression codecs. In particular, CompressAI includes pre-trained models and evaluation tools to compare learned methods with traditional codecs. Multiple models from the state-of-the-art on learned end-to-end compression have thus been reimplemented in PyTorch and trained from scratch. We also report objective comparison results using PSNR and MS-SSIM metrics vs. bit-rate, using the Kodak image dataset as test set. Although this framework currently implements models for still-picture compression, it is intended to be soon extended to the video compression domain.
19 pages, 11 figures
References in corpus (2)
Cited by in corpus (6)
- COIN: COmpression with Implicit Neural representations
- A Cross Channel Context Model for Latents in Deep Image Compression
- Enhanced Invertible Encoding for Learned Image Compression
- Variable-Rate Deep Image Compression through Spatially-Adaptive Feature Transform
- A Combined Deep Learning based End-to-End Video Coding Architecture for YUV Color Space
- Learned Video Compression with Residual Prediction and Loop Filter