arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.05810eess.IVcs.CV
收藏

两阶段就足够了:一种用于灵活视频压缩感知的简明深度展开重建网络

Two-Stage is Enough: A Concise Deep Unfolding Reconstruction Network for Flexible Video Compressive Sensing

Computer Network Information Center, Chinese Academy of Science(中国科学院计算机网络信息中心);University of Chinese Academy of Sciences(中国科学院大学);Westlake University(西湖大学)

Siming Zheng, Xiaoyu Yang, Xin Yuan

摘要:

本文研究了深度展开/滚动结构下的视频压缩感知重构问题。然而,我们的目标是建立一个灵活和简洁的模型,使用最少的阶段。与现有的用于逆问题的深度展开网络不同,其中使用更多的阶段以获得更高的性能,但对不同的掩码和尺度没有灵活性,因此,我们证明了2阶段深度展开网络可以导致最先进的(SOTA)结果(与单阶段模型相比,PSNR增益为1.7dB,RevSCI)。由于深度展开的优点,所提出的方法具有适应新掩码的特性,并且可以扩展到大数据而无需任何额外的训练。此外,我们扩展了所提出的模型进行联合重建和去马赛克。实验结果表明,本文提出的两阶段模型也实现了SOTA彩色图像重建,与基于即插即用框架的SOTA算法相比,PSNR提高了2.3dB以上,重建速度提高了17倍以上。此外,我们还发现,我们的网络也是灵活的掩模调制和规模大小的颜色重建,使一个单一的训练网络可以应用于不同的硬件系统。代码和模型将向公众发布。

英文摘要:

We consider the reconstruction problem of video compressive sensing (VCS) under the deep unfolding/rolling structure. Yet, we aim to build a flexible and concise model using minimum stages. Different from existing deep unfolding networks used for inverse problems, where more stages are used for higher performance but without flexibility to different masks and scales, hereby we show that a 2-stage deep unfolding network can lead to the state-of-the-art (SOTA) results (with a 1.7dB gain in PSNR over the single stage model, RevSCI) in VCS. The proposed method possesses the properties of adaptation to new masks and ready to scale to large data without any additional training thanks to the advantages of deep unfolding. Furthermore, we extend the proposed model for color VCS to perform joint reconstruction and demosaicing. Experimental results demonstrate that our 2-stage model has also achieved SOTA on color VCS reconstruction, leading to a >2.3dB gain in PSNR over the previous SOTA algorithm based on plug-and-play framework, meanwhile speeds up the reconstruction by >17 times. In addition, we have found that our network is also flexible to the mask modulation and scale size for color VCS reconstruction so that a single trained network can be applied to different hardware systems. The code and models will be released to the public.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑