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

PDCFNet:通过像素差分卷积增强水下图像

PDCFNet: Enhancing Underwater Images through Pixel Difference Convolution

China Agricultural University(中国农业大学)

Song Zhang, Daoliang Li, Ran Zhao

AI总结:

本文提出PDCFNet,通过像素差分卷积捕获高频特征并结合跨层级特征融合,显著提升水下图像增强性能,在UIEB数据集上取得最佳PSNR和SSIM。

摘要:

大多数深度学习方法利用普通卷积来增强水下图像。虽然普通卷积在捕获局部特征和学习图像的空间层次结构方面表现出色,但它倾向于平滑输入图像,这在一定程度上限制了特征表达和建模。水下退化图像的一个显著特征是模糊,增强的目标是使图像中的纹理和细节(高频特征)更加清晰可见。因此,我们认为利用高频特征可以提高增强性能。为了解决这个问题,我们引入了像素差分卷积(PDC),它专注于图像中具有显著变化的梯度信息,从而改善对增强图像的建模。我们提出了一种基于PDC和跨层级特征融合的水下图像增强网络PDCFNet。具体来说,我们设计了一个基于PDC的细节增强模块,该模块采用并行PDC来捕获高频特征,从而实现更好的细节和纹理增强。所设计的跨层级特征融合模块对来自不同层级的特征执行拼接和乘法等操作,确保不同特征之间充分的交互和增强。我们提出的PDCFNet在UIEB数据集上实现了27.37的PSNR和92.02的SSIM,达到了迄今为止的最佳性能。我们的代码可在https://github.com/zhangsong1213/PDCFNet获取。

英文摘要:

Majority of deep learning methods utilize vanilla convolution for enhancing underwater images. While vanilla convolution excels in capturing local features and learning the spatial hierarchical structure of images, it tends to smooth input images, which can somewhat limit feature expression and modeling. A prominent characteristic of underwater degraded images is blur, and the goal of enhancement is to make the textures and details (high-frequency features) in the images more visible. Therefore, we believe that leveraging high-frequency features can improve enhancement performance. To address this, we introduce Pixel Difference Convolution (PDC), which focuses on gradient information with significant changes in the image, thereby improving the modeling of enhanced images. We propose an underwater image enhancement network, PDCFNet, based on PDC and cross-level feature fusion. Specifically, we design a detail enhancement module based on PDC that employs parallel PDCs to capture high-frequency features, leading to better detail and texture enhancement. The designed cross-level feature fusion module performs operations such as concatenation and multiplication on features from different levels, ensuring sufficient interaction and enhancement between diverse features. Our proposed PDCFNet achieves a PSNR of 27.37 and an SSIM of 92.02 on the UIEB dataset, attaining the best performance to date. Our code is available at https://github.com/zhangsong1213/PDCFNet.

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

↑