高级检索

基于频域特征引导的雾霾数据域迁移模型

A Fog and Haze Data Domain Transfer Model Based on Style-Content Decoupling

  • 摘要: 针对训练域与真实域雾霾存在域间差异的问题,提出一种基于频域特征引导的雾霾数据域迁移模型。首先,基于小波变换分别提取清晰图像的内容特征和有雾图像的雾霾特征,并通过交叉注意力对齐;其次,通过提出的深度一致性损失和色彩一致性损失对域迁移模型进行优化,使其将有雾图像的雾霾特征迁移到清晰图像上;最后,合成大量的雾霾图像,扩充训练域雾霾数据,弥补域间差异,提升去雾模型的能力。在O-HAZE、NH-HAZE和OTS数据集上的充分的实验结果表明,所提模型能够提升现有去雾方法在真实场景中的泛化能力,且在O-HAZE数据集上,相较于次优的方法分别PSNR和SSIM分别提升了1.13dB和0.005 4。

     

    Abstract: To address the domain gap between synthetic hazy data used for training and real-world haze, we propose a frequency-domain feature–guided haze data domain translation model. Specifically, content features from clear images and haze-related features from hazy images are separately extracted via wavelet transform and aligned through a cross-attention mechanism. The proposed domain translation model is further optimized using a depth consistency loss and a color consistency loss, enabling effective transfer of haze characteristics from hazy images to clear ones. Based on the translated results, a large-scale hazy dataset is synthesized to augment the training domain, thereby reducing domain discrepancy and enhancing the performance of dehazing models. Extensive experiments on the O-HAZE, NH-HAZE, and OTS datasets demonstrate that the proposed approach significantly improves the generalization ability of existing dehazing methods in real-world scenarios. In particular, on the O-HAZE dataset, our method outperforms the second-best approach by 1.13 dB in PSNR and 0.005 4 in SSIM.

     

/

返回文章
返回