高级检索

基于渐进式频域学习的图像去雨算法

Progressive Frequency Domain Learning Based Image Deraining Algorithm

  • 摘要: 针对现有深度学习去雨算法在困难样本上雨水去除不彻底、纹理细节易丢失的问题,提出一种基于渐进式频域学习的图像去雨算法。该算法由2个结构相同的U-Net级联而成:首先,输入图像经低频校正U-Net获得初步去雨结果,通过标签图像在空间域施加逐像素约束,并在小波域施加低频约束;随后,初步结果输入高频精修U-Net,输出最终去雨结果,在空间域逐像素约束与小波域高频约束的基础上,额外引入有雨图像的傅里叶频域对比学习约束。此外,为U-Net设计了一种新颖的空间-频域双分支架构:一方面采用自适应稀疏自注意力机制提取空间域特征并抑制冗余干扰;另一方面利用多尺度傅里叶调制模块补充频域信息。在4个公开数据集上的实验表明,所提算法的平均PSNR和SSIM指标相较于现有对比算法分别提高1.10%和0.27%,在多样化场景下实现了显著的性能提升。

     

    Abstract: To address the limitations of incomplete rain streak removal and easy loss of texture details on hard sam-ples in existing deep learning based image deraining algorithms, this paper proposes an image deraining algorithm based on progressive frequency domain learning. The algorithm is composed of two cascaded U-Nets with identical structures. First, the input image is processed by the low-frequency correction U-Net to obtain a preliminary deraining result. A pixel-wise constraint is imposed in the spatial domain using the ground truth image, and a low-frequency constraint is applied in the wavelet domain. Subsequently, the preliminary result is fed into the high-frequency refinement U-Net to generate the final deraining output. On the basis of the pixel-wise constraint in the spatial domain and the high-frequency constraint in the wavelet domain, an additional contrastive constraint in Fourier domain is introduced. Furthermore, a novel spatial-frequency dual branch architecture is designed for the U-Nets. On one hand, an adaptive sparse self-attention mechanism is adopted to extract spatial domain features and suppress redundant interference. On the other hand, a multi-scale Fourier modulator is utilized to supplement frequency domain information. Experiments conducted on four public datasets demonstrate that the average PSNR and SSIM metrics of the proposed algorithm are improved by 1.10% and 0.27% respectively compared with existing state-of-the-art algorithms, achieving significant performance gains across diverse scenarios.

     

/

返回文章
返回