Python(pytorch) code for the paper: Siming Zheng, Yang Liu, Ziyi Meng, Mu Qiao, Zhishen Tong, Xiaoyu Yang, Shensheng Han, and Xin Yuan, "Deep plug-and-play priors for spectral snapshot compressive imaging," Photon. Res. 9, B18-B29 (2021)[pdf]
We propose a plug-and-play (PnP) method, which uses deep-learning-based denoisersas regularization priors for spectral snapshot compressive imaging (SCI). Our method is efficient in terms of reconstruction quality and speed trade-off, and flexible to be ready-to-use for differentcompressive coding mechanisms.
English:
This repository is mainly used to fix some bugs in the original code from the paper's authors. All modifications can be found in the dev branch.
Chinese:
本仓库主要用于修复原论文作者代码的一些BUG,所有改动位于dev 分支下。
@article{Zheng:21,
author = {Siming Zheng and Yang Liu and Ziyi Meng and Mu Qiao and Zhishen Tong and Xiaoyu Yang and Shensheng Han and Xin Yuan},
journal = {Photon. Res.},
keywords = {Compressive imaging; Hyperspectral imaging; Multispectral imaging; Reconstruction algorithms; Remote sensing; Spatial light modulators},
number = {2},
pages = {B18--B29},
publisher = {OSA},
title = {Deep plug-and-play priors for spectral snapshot compressive imaging},
volume = {9},
month = {Feb},
year = {2021},
url = {http:https://www.osapublishing.org/prj/abstract.cfm?URI=prj-9-2-B18},
doi = {10.1364/PRJ.411745},
abstract = {We propose a plug-and-play (PnP) method that uses deep-learning-based denoisers as regularization priors for spectral snapshot compressive imaging (SCI). Our method is efficient in terms of reconstruction quality and speed trade-off, and flexible enough to be ready to use for different compressive coding mechanisms. We demonstrate the efficiency and flexibility in both simulations and five different spectral SCI systems and show that the proposed deep PnP prior could achieve state-of-the-art results with a simple plug-in based on the optimization framework. This paves the way for capturing and recovering multi- or hyperspectral information in one snapshot, which might inspire intriguing applications in remote sensing, biomedical science, and material science. Our code is available at: https://github.com/zsm1211/PnP-CASSI.},
}