CompressAI

CompressAI (compress-ay) is a PyTorch library and evaluation platform for end-to-end compression research.

• Introduction
• Installation

Tutorials

• Training
• Custom model

Library API

• compressai
• compressai.ans
• compressai.datasets
• compressai.entropy_models
• compressai.latent_codecs
• compressai.layers
• compressai.losses
• compressai.models
• compressai.ops
• compressai.transforms

Model Zoo

• Image compression
  • Training
  • Models
  • Performances
• Video compression
  • Models

Utils

• Command line usage
  • bench
  • eval_model
  • find_close
  • plot
  • update_model

Development

• Github repository