InterDigitalInc/CompressAI
A PyTorch library and evaluation platform for end-to-end compression research
What it solves
它提供一个标准化的基于 PyTorch 的框架，用于研究和开发端到端数据压缩（使用深度学习），让研究者能够超越传统编解码器，创建学习式的图像和视频压缩模型。
How it works
CompressAI 提供一套专为学习式压缩设计的自定义 PyTorch 操作、层和模型。它包含 TensorFlow compression 库的部分移植，并提供可直接使用的预训练模型，用于编码和解码。库中还包括带有率失真损失的训练脚本以及用于将学习模型与传统编解码器（如 BPG、VTM、x265）性能进行比较的评估工具。
Who it’s for
适用于从事数据压缩、计算机视觉和深度学习的 AI 研究人员和工程师，需要一个平台来训练、评估和基准测试学习式图像和视频压缩模型。
Highlights
- End-to-End Research Platform: 从训练到评估的完整流水线，支持深度学习式压缩。
- Pre-trained Model Zoo: 可直接使用的学习式图像压缩预训练模型集合。
- Benchmarking Tools: 内置脚本可将学习模型与传统图像和视频压缩标准进行比较。
- Flexible Training: 支持自定义模型实现和跨多 GPU 的分布式训练。