Skip to content

Official Pytorch Implementation of Bidirectional Stereo Image Compression with Cross-Dimensional Entropy Model [ECCV'24]

Notifications You must be signed in to change notification settings

LIUZhening111/BiSIC

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

[ECCV'24] BiSIC: Bidirectional Stereo Image Compression

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

[ArXiv Preprint]

⭐Overview

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.

Model architecture

✨Performance Illustration

We illustrate the R-D performances and qualitative visualization of our method. For more details, please check the official paper.

R-D performance on InStereo2K dataset

Compression results visualization

😋Pretrained Model

We provide pre-trained checkpoints through Google Drive Link. You can inference the model through the bash files provided in the folder /script.

❤️Acknowledgement

We would like to appreciate the following fantastic open-source works contributing to the implementation in image compression community ❤️❤️ :

👉Citation

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}
}

About

Official Pytorch Implementation of Bidirectional Stereo Image Compression with Cross-Dimensional Entropy Model [ECCV'24]

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published