A PyTorch library and evaluation platform for end-to-end compression research
CompressAI CompressAI (compress-ay) is a PyTorch library and evaluation platform for end-to-end compression research. CompressAI currently provides: custom operations, layers and models for deep learning based data compression a partial port of the official TensorFlow compression library pre-trained end-to-end compression models for learned image compression evaluation scripts to compare learned models against classical image/videocompression codecs Note: Multi-GPU support is now experimental. Installation CompressAI supports python 3.6+ and PyTorch 1.4+. pip: pip install compressai Note: wheels are available for Linux and […]
Read more