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 的分散式訓練。