CompressAI：一个用于端到端压缩研究的 PyTorch 库与评测平台
计算机视觉与模式识别
2020-11-06 v1 图像与视频处理
摘要
本文介绍 CompressAI，一个提供自定义算子、层、模型和工具以研究、开发和评测端到端图像与视频压缩编解码器的平台。特别地，CompressAI 包含预训练模型与评测工具，用于将学习方法与传统编解码器进行比较。因此，最先进的端到端压缩中的多个模型已在 PyTorch 中重新实现并从头训练。我们还报告了以 Kodak 图像数据集为测试集，使用 PSNR 与 MS-SSIM 指标相对于码率的客观比较结果。尽管该框架目前实现了静态图像压缩模型，但其旨在很快扩展至视频压缩领域。
引用
@article{arxiv.2011.03029,
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 preprint arXiv:2011.03029},
year = {2020}
}
备注
19 pages, 11 figures