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

TSNet:一种采用多尺度融合与自适应学习的两阶段图像去雾网络

TSNet:A Two-stage Network for Image Dehazing with Multi-scale Fusion and Adaptive Learning

Tianjin University(天津大学)

Xiaolin Gong, Zehan Zheng, Heyuan Du

AI总结:

本文提出两阶段图像去雾网络 TSNet,通过 MSFM 多尺度融合与 ALM 自适应学习增强泛化,并由第二阶段修复伪影和颜色失真;同时改用反向雾度图作为学习目标,在合成与真实数据集上性能优于现有方法。

摘要:

长期以来,图像去雾一直是一个热门研究课题。以往基于深度学习的图像去雾方法未能在合成数据集和真实世界数据集上同时取得令人满意的去雾效果,表现出较差的泛化能力。此外,单阶段网络往往会在输出图像的许多区域产生伪影和颜色失真。为解决这些问题,本文提出了一种名为 TSNet 的两阶段图像去雾网络,主要由多尺度融合模块(MSFM)和自适应学习模块(ALM)组成。具体而言,MSFM 和 ALM 增强了 TSNet 的泛化能力。MSFM 能够在多个尺度上获得大感受野,并整合不同频率的特征,以减小输入与学习目标之间的差异。ALM 能够主动学习图像中的感兴趣区域,并更有效地恢复纹理细节。此外,TSNet 被设计为两阶段网络:第一阶段网络执行图像去雾,第二阶段网络用于改善第一阶段网络结果中存在的伪影和颜色失真等问题。我们还将学习目标从真值图像改为反向雾度图,从而提高了 TSNet 的学习效率。大量实验表明,与以往最先进的方法相比,TSNet 在合成数据集和真实世界数据集上均展现出更优的去雾性能。

英文摘要:

Image dehazing has been a popular topic of research for a long time. Previous deep learning-based image dehazing methods have failed to achieve satisfactory dehazing effects on both synthetic datasets and real-world datasets, exhibiting poor generalization. Moreover, single-stage networks often result in many regions with artifacts and color distortion in output images. To address these issues, this paper proposes a two-stage image dehazing network called TSNet, mainly consisting of the multi-scale fusion module (MSFM) and the adaptive learning module (ALM). Specifically, MSFM and ALM enhance the generalization of TSNet. The MSFM can obtain large receptive fields at multiple scales and integrate features at different frequencies to reduce the differences between inputs and learning objectives. The ALM can actively learn of regions of interest in images and restore texture details more effectively. Additionally, TSNet is designed as a two-stage network, where the first-stage network performs image dehazing, and the second-stage network is employed to improve issues such as artifacts and color distortion present in the results of the first-stage network. We also change the learning objective from ground truth images to opposite fog maps, which improves the learning efficiency of TSNet. Extensive experiments demonstrate that TSNet exhibits superior dehazing performance on both synthetic and real-world datasets compared to previous state-of-the-art methods.

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

↑