Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Self-supervised Neural Networks for Spectral Snapshot Compressive Imaging

About

We consider using {\bf\em untrained neural networks} to solve the reconstruction problem of snapshot compressive imaging (SCI), which uses a two-dimensional (2D) detector to capture a high-dimensional (usually 3D) data-cube in a compressed manner. Various SCI systems have been built in recent years to capture data such as high-speed videos, hyperspectral images, and the state-of-the-art reconstruction is obtained by the deep neural networks. However, most of these networks are trained in an end-to-end manner by a large amount of corpus with sometimes simulated ground truth, measurement pairs. In this paper, inspired by the untrained neural networks such as deep image priors (DIP) and deep decoders, we develop a framework by integrating DIP into the plug-and-play regime, leading to a self-supervised network for spectral SCI reconstruction. Extensive synthetic and real data results show that the proposed algorithm without training is capable of achieving competitive results to the training based networks. Furthermore, by integrating the proposed method with a pre-trained deep denoising prior, we have achieved state-of-the-art results. {Our code is available at \url{https://github.com/mengziyi64/CASSI-Self-Supervised}.}

Ziyi Meng, Zhenming Yu, Kun Xu, Xin Yuan• 2021

Related benchmarks

TaskDatasetResultRank
HSI ReconstructionKAIST 10 scenes (Scene2)
PSNR25.89
39
Hyperspectral Image ReconstructionKAIST 10 simulation scenes (test)
PSNR31.26
30
Hyperspectral Image ReconstructionKAIST simulation (Average test)
PSNR31.26
26
HSI ReconstructionKAIST 10 scenes (Scene5)
PSNR27.45
25
Video ReconstructionRunner
PSNR27.92
21
Video ReconstructionAerial
PSNR24.74
21
Video ReconstructionAverage
PSNR25.25
21
Video ReconstructionCrash
PSNR23.46
21
Video ReconstructionKobe
PSNR22.47
21
Video ReconstructionTraffic
PSNR19.95
21
Showing 10 of 27 rows

Other info

Follow for update