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.
python 3.7
compressai 1.2.0
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.pyhttps://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.