Hugging Peers

CompressAI: a PyTorch library and evaluation platform for end-to-end compression research

☆ Save
Jean Bégaint,Fabien Racapé,Simon Feltman,Akshay Pushparaja
arXiv (Cornell University) Advanced Data Compression TechniquesVideo Coding and Compression TechnologiesAdvanced Neural Network Applicationscs.CVeess.IV
288 citations published

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.

Cite this paper

@article{bgaint2020,
  title = {CompressAI: a PyTorch library and evaluation platform for end-to-end compression research},
  author = {Jean Bégaint and Fabien Racapé and Simon Feltman and Akshay Pushparaja},
  journal = {arXiv (Cornell University)},
  year = {2020},
  month = {Nov},
  doi = {10.48550/arxiv.2011.03029},
  url = {https://doi.org/10.48550/arxiv.2011.03029},
}