arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2212.11473cs.CV

利用层次对比学习恢复雾霾天气下的视觉

Restoring Vision in Hazy Weather with Hierarchical Contrastive Learning

Nanjing University(南京大学);Memorial University of Newfoundland(纽芬兰纪念大学);Australian National University(澳大利亚国立大学);Sun Yat-sen University(中山大学);Wenzhou University(温州大学)

Tao Wang, Guangpin Tao, Wanglong Lu, Kaihao Zhang, Wenhan Luo, Xiaoqin Zhang, Tong Lu

AI总结:

本文提出层次对比去雾方法HCD,通过层次交互模块和层次对比损失充分利用特征层次关系,在多个公开数据集上取得优于现有方法的去雾效果。

摘要:

雾霾天气条件下的图像恢复,即单图像去雾,对多种计算机视觉应用一直具有重要意义。近年来,基于深度学习的方法取得了成功。然而,现有的图像去雾方法通常忽视神经网络中特征的层次性,未能充分挖掘特征之间的关系。为此,我们提出了一种有效的图像去雾方法,名为层次对比去雾(HCD),该方法基于特征融合和对比学习策略。HCD由一个层次去雾网络(HDN)和一个新颖的层次对比损失(HCL)组成。具体而言,HDN的核心设计是一个层次交互模块,它利用多尺度激活来逐层修正特征响应。为配合HDN的训练,我们提出了HCL,它在层次配对的样本上进行对比学习,以促进雾霾去除。在公开数据集RESIDE、HazeRD和DENSE-HAZE上的大量实验表明,HCD在PSNR、SSIM方面定量优于最先进的方法,并取得了更好的视觉质量。

英文摘要:

Image restoration under hazy weather condition, which is called single image dehazing, has been of significant interest for various computer vision applications. In recent years, deep learning-based methods have achieved success. However, existing image dehazing methods typically neglect the hierarchy of features in the neural network and fail to exploit their relationships fully. To this end, we propose an effective image dehazing method named Hierarchical Contrastive Dehazing (HCD), which is based on feature fusion and contrastive learning strategies. HCD consists of a hierarchical dehazing network (HDN) and a novel hierarchical contrastive loss (HCL). Specifically, the core design in the HDN is a hierarchical interaction module, which utilizes multi-scale activation to revise the feature responses hierarchically. To cooperate with the training of HDN, we propose HCL which performs contrastive learning on hierarchically paired exemplars, facilitating haze removal. Extensive experiments on public datasets, RESIDE, HazeRD, and DENSE-HAZE, demonstrate that HCD quantitatively outperforms the state-of-the-art methods in terms of PSNR, SSIM and achieves better visual quality.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑