高级检索

    基于改进U-net的哈达玛单像素成像的重建方法

    Hadamard Single-Pixel Imaging Based on Improved U-net

    • 摘要: 单像素成像因其成本低、灵活性和抗干扰性强等优点, 在众多领域都有广泛应用。然而常用的正交基函数, 如傅里叶、小波、哈达玛等无法充分捕捉复杂图像的特征和变换过程中频谱系数的丢失, 导致重建质量不佳。为了提升基于哈达玛变换的单像素成像的成像质量,提出了一种基于改进U-net算法的哈达玛单像素成像重建方法,该算法结合哈达玛单像素成像的频谱特性构造损失函数,同时引入自适应池化和注意力机制来强化输入特征,通过通道注意力和空间注意力机制,动态增强模型对重要特征的响应。实验结果表明,在6.25%的低采样率下,与传统哈达玛单像素成像butterfly样本相比,峰值信噪比(PSNR)提升约12%,结构相似度(SSIM)提升约30%;与基于SRCNN的哈达玛单像素成像方法相比,PSNR提升约4%,结构相似度(SSIM)提升约3%,可以实现超低采样率下的单像素成像的良好重建任务,实现了快速高质量的单像素成像。

       

      Abstract: Single pixel imaging has been widely used in many fields due to its advantages of low cost, flexibility, and strong anti-interference ability. However, commonly used orthogonal basis functions such as Fourier, wavelet, Hadamard, etc. cannot fully capture the features of complex images and the loss of spectral coefficients during the transformation process, resulting in poor reconstruction quality. In order to improve the imaging quality of single pixel imaging based on Hadamard transform, this paper proposes a Hadamard single pixel imaging reconstruction method based on an improved U-net algorithm. The algorithm combines the spectral characteristics of Hadamard single pixel imaging to construct a loss function, and introduces adaptive pooling and attention mechanisms to enhance input features. Through channel attention and spatial attention mechanisms, the model's response to important features is dynamically enhanced. The experimental results show that at a low sampling rate of 6.25%, compared with traditional Hadamard single pixel imaging of butterfly samples, the peak signal-to-noise ratio (PSNR) is improved by about 12%, and the structural similarity (SSIM) is improved by about 30%. Compared with the Hadamard single pixel imaging method based on SRCNN, the PSNR is improved by about 4% and the structural similarity (SSIM) is improved by about 3%.It can achieve good reconstruction tasks for single pixel imaging at ultra-low sampling rates.

       

    /

    返回文章
    返回