论文标题

基于块压缩感和深度学习的单像素图​​像重建

Single-Pixel Image Reconstruction Based on Block Compressive Sensing and Deep Learning

论文作者

Lau, Stephen L. H., Chong, Edwin K. P.

论文摘要

单像素成像(SPI)是一种新型成像技术,其工作原理基于压缩感(CS)理论。在SPI中,数据是通过一系列压缩测量获得的,并重建了相应的图像。通常,重建算法(例如基础追求)依赖图像中的稀疏性假设。但是,深度学习的最新进展发现了其在重建CS图像中的用途。尽管在模拟中显示出有希望的结果,但尚不清楚如何在实际的SPI设置中实现这种算法。在本文中,我们证明了对SPI图像的重建以及块压缩感(BCS)的重建。我们还提出了一种基于卷积神经网络的新型重建模型,该模型优于其他竞争性CS重建算法。此外,通过将BC纳入我们的深度学习模型中,我们能够重建以上图像大小以上的任何大小的图像。此外,我们表明我们的模型能够重建从SPI设置获得的图像,同时接受了自然图像的训练,这可能与SPI图像大不相同。这为验证的深度学习模型的可行性打开了机会,以重建来自各个领域的图像。

Single-pixel imaging (SPI) is a novel imaging technique whose working principle is based on the compressive sensing (CS) theory. In SPI, data is obtained through a series of compressive measurements and the corresponding image is reconstructed. Typically, the reconstruction algorithm such as basis pursuit relies on the sparsity assumption in images. However, recent advances in deep learning have found its uses in reconstructing CS images. Despite showing a promising result in simulations, it is often unclear how such an algorithm can be implemented in an actual SPI setup. In this paper, we demonstrate the use of deep learning on the reconstruction of SPI images in conjunction with block compressive sensing (BCS). We also proposed a novel reconstruction model based on convolutional neural networks that outperforms other competitive CS reconstruction algorithms. Besides, by incorporating BCS in our deep learning model, we were able to reconstruct images of any size above a certain smallest image size. In addition, we show that our model is capable of reconstructing images obtained from an SPI setup while being priorly trained on natural images, which can be vastly different from the SPI images. This opens up opportunity for the feasibility of pretrained deep learning models for CS reconstructions of images from various domain areas.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源