高级检索

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

Memory Mechanism and Frequency Learning for Unpaired Image Dehazing

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

     

    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.29dB and 0.19dB, respectively.

     

/

返回文章
返回