Awesome
About
An unofficial pytorch implementation of CVPR2021 paper "Checkerboard Context Model for Efficient Learned Image Compression".
This project is based on CompressAI, "mbt2018 + checkerboard" is implemented in model.py
.
Usage
enviroment
python 3.7
compressai 1.2.0
demo
Due to the limitation of file size in github, you should download checkpoint from Google drive, and then put it into the project fold.
update: Sorry, the checkpoint is lost for some mistakes, please retrain the model using compressai.
pip install compressai
python demo.py
Reference
https://github.com/JiangWeibeta/Checkerboard-Context-Model-for-Efficient-Learned-Image-Compression
https://github.com/InterDigitalInc/CompressAI
https://github.com/huzi96/Coarse2Fine-PyTorch
Paper: https://arxiv.org/abs/2103.15306
See my blog for more details.