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 にまたがる分散トレーニングをサポート。