A PyTorch library and evaluation platform for end-to-end compression research

This paper presents CompressAI, an open-source library that provides custom operations, layers, models and tools to research, develop, and evaluate end-to-end image and video codecs. In particular, CompressAI includes pre-trained models and evaluation tools to compare learned methods with traditional codecs. Multiple models from the state-of-the-art on learned end-to-end image compression have been reimplemented in PyTorch [1] and trained from scratch using the Vimeo90K2 training dataset [2].