Code for Effective Snapshot Compressive-spectral Imaging via Deep Denoising and Total Variation Priors (CVPR 2021).
- python3
- pytorch, cvxpy, scipy, numpy, skimage, tqdm
- run our algorithm
python pnp_gap_HSI_tv_ffdnet.py- run two baseline algorithms
# Run plug-and-play gap based on 3d TV denoiser
python pnp_gap_HSI_3dtv.py
# Run plug-and-play gap based on FFDNet
python pnp_gap_HSI_ffdnet.py