Deep Motion Regularizer for Video Snapshot Compressive Imaging
Zan Chen, Ran Li, Yongqiang Li, Yuanjing Feng, Xingsong Hou, Xueming Qian
Abstract
Video snapshot compressive imaging (SCI) samples 3D high-speed video frames with temporally varying spatial modulation and compresses them into a single 2D measurement, and the SCI reconstruction algorithm aims to recover the original high-speed frames from the measurement. However, conventional video SCI systems encounter challenges when raising the frame rate of the reconstructed video. To achieve higher frame rates, the modulation mask's rate must be increased, which in turn leads to an increase in the associated hardware expenses. In this paper, we propose a deep unfolding-based reconstruction framework with optical flow for video SCI. The framework recovers both observed and unobserved frames from measurements, resulting in increased video frame rate. To estimate the optical flow, we transform the video features of the network into optical flow features during the iteration. Then, we design a deep denoiser and an optical flow-based motion regularizer combined with the voxels of coarse reconstructed frames to update the observed and unobserved frames. To improve the performance, we employ group convolution in the network and fuse the optical flow information from different phases to reduce the information loss. We further extend the proposed deep unfolding framework to the reconstruction of color SCI videos. Extensive experiments on benchmark data and real data prove that our proposed method has state-of-the-art reconstruction performance and visual effects.