高级检索

基于记忆机制和频域学习的非配对图像去雾算法

Memory Mechanism and Frequency Learning for Unpaired Image Dehazing

  • 摘要: 主流的图像去雾算法在训练时对高质量像素级配对数据的依赖程度大, 实用能力较弱. 鉴于记忆机制能够在训练过程中学习并存储大量有价值的先验知识辅助特征学习, 且雾霾相关的退化主要蕴含在频域的幅度谱的性质. 本文提出了一种基于频域记忆增强的非配对图像去雾算法(FMA-Net), 该算法通过在去雾过程增强频域特征中的幅度谱来实现有效的雾霾去除.首先, 本论文提出了训练时使用的记忆特征学习模块(MLM), 通过从非配对的无雾图像中预先学习无雾图像幅度谱特征来更新记忆特征.基于此, 本论文还提出了记忆增强模块(MAM), 使用硬注意力机制来表征记忆特征和雾霾图像幅度谱特征之间的关系, 并通过重建新的幅度谱来去除输入图像中的雾霾. 此外, 本文还将MAM配备在多个不同尺度的特征上来提升算法复原出清晰图像的能力.相关实验结果表明所提出的FMA-Net在包含合成的、人造的和真实的雾霾的五个基准测试集上的量化指标和视觉质量均优于多种当前先进算法.其中在SOTS和HSTS-S两个代表性基准上的PSNR分别提高了0.29dB和0.19dB.

     

    Abstract: Mainstream image dehazing algorithms rely heavily on high-quality pixel-level paired data during training, limiting their practical capabilities. Motivated by the fact that the memory mechanism can learn and store a large amount of valuable prior knowledge during the training process to assist the feature learning, and the haze-related degradation is mainly implied in the amplitude spectrum. This paper proposes an unpaired image dehazing method based on frequency memory-augmentation (FMA-Net), which achieves effective haze removal by augmenting the amplitude spectrum in the frequency domain. First, this paper proposes the memory feature learning module (MLM) used during training to update the memory features by pre-learning the amplitude spectrum features from unpaired haze-free images. Based on this, this paper also proposes a memory augmentation module (MAM), which uses a hard-attention mechanism to represent the relationship between the memories and the amplitude spectrum features of the haze image, and removes the haze by reconstructing a new amplitude spectrum. In addition, the MAM can be equipped with multiple scales to improve the algorithm's ability. Experimental results show that the proposed FMA-Net outperforms the state-of-the-art algorithms in terms of both quantitative and visual quality on five benchmark test sets containing synthetic, artificial and real haze. Among them, the PSNR on two representative benchmarks, SOTS and HSTS-S, is improved by 0.29 dB and 0.19 dB, respectively.

     

/

返回文章
返回