This is the official PyTorch implementation of our ECCV 2024 paper:
Bidirectional Stereo Image Compression with Cross-Dimensional Entropy Model
Zhening Liu, Xinjie Zhang, Jiawei Shao, Zehong Lin, Jun Zhang
With the rapid advancement of stereo vision technologies, stereo image compression has emerged as a crucial field that continues to draw significant attention. Previous approaches have primarily employed a unidirectional paradigm, where the compression of one view is dependent on the other, resulting in imbalanced compression. To address this issue, we introduce a symmetric bidirectional stereo image compression architecture, named BiSIC. Specifically, we propose a 3D convolution based codec backbone to capture local features and incorporate bidirectional attention blocks to exploit global features. Moreover, we design a novel cross-dimensional entropy model that integrates various conditioning factors, including the spatial context, channel context, and stereo dependency, to effectively estimate the distribution of latent representations for entropy coding. Extensive experiments demonstrate that our proposed BiSIC outperforms conventional image/video compression standards, as well as state-of-the-art learning-based methods, in terms of both PSNR and MS-SSIM.
We illustrate the R-D performances and qualitative visualization of our method. For more details, please check the official paper.
We provide pre-trained checkpoints through Google Drive Link. You can inference the model through the bash files provided in the folder /script
.
We would like to appreciate the following fantastic open-source works contributing to the implementation in image compression community ❤️❤️ :
- CompressAI [related paper]
- Checkerboard Context Model for Efficient Learned Image Compression [related paper]
- SASIC [related paper]
- LDMIC [related paper]
If any parts of our paper and code help your research, please consider citing us and giving a star to our repository.
@article{liu2024bidirectional,
title={Bidirectional Stereo Image Compression with Cross-Dimensional Entropy Model},
author={Liu, Zhening and Zhang, Xinjie and Shao, Jiawei and Lin, Zehong and Zhang, Jun},
journal={arXiv preprint arXiv:2407.10632},
year={2024}
}