arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2309.04089cs.CV

面向充分空间-频率交互的梯度感知水下图像增强

Toward Sufficient Spatial-Frequency Interaction for Gradient-aware Underwater Image Enhancement

Nanjing Normal University(南京师范大学)

Chen Zhao, Weiling Cai, Chenyu Dong, Ziqi Zeng

AI总结:

针对水下图像退化问题,提出两阶段SFGNet,通过DSFFNet实现空间-频率充分交互,并利用GAC依据梯度图增强细节与结构,实验取得竞争性性能。

摘要:

水下图像遭受复杂多样的退化,这不可避免地影响水下视觉任务的性能。然而,现有的大多数基于学习的水下图像增强(UIE)方法主要在空间域恢复此类退化,很少关注傅里叶频率信息。本文提出一种基于空间-频率交互和梯度图的新型UIE框架SFGNet,该框架包含两个阶段。具体而言,第一阶段提出密集空间-频率融合网络(DSFFNet),主要包括所设计的密集傅里叶融合块和密集空间融合块,通过这两个块之间的交叉连接实现充分的空间-频率交互。第二阶段提出梯度感知校正器(GAC),通过梯度图进一步增强图像的感知细节和几何结构。在两个真实世界水下图像数据集上的实验结果表明,该方法能够成功增强水下图像,并在视觉质量提升方面取得具有竞争力的性能。代码可在https://github.com/zhihefang/SFGNet获取。

英文摘要:

Underwater images suffer from complex and diverse degradation, which inevitably affects the performance of underwater visual tasks. However, most existing learning-based Underwater image enhancement (UIE) methods mainly restore such degradations in the spatial domain, and rarely pay attention to the fourier frequency information. In this paper, we develop a novel UIE framework based on spatial-frequency interaction and gradient maps, namely SFGNet, which consists of two stages. Specifically, in the first stage, we propose a dense spatial-frequency fusion network (DSFFNet), mainly including our designed dense fourier fusion block and dense spatial fusion block, achieving sufficient spatial-frequency interaction by cross connections between these two blocks. In the second stage, we propose a gradient-aware corrector (GAC) to further enhance perceptual details and geometric structures of images by gradient map. Experimental results on two real-world underwater image datasets show that our approach can successfully enhance underwater images, and achieves competitive performance in visual quality improvement. The code is available at https://github.com/zhihefang/SFGNet.

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

↑