Underwater image enhancement (UIE) is a challenging task due to the complex degradation caused by underwater environments. To solve this issue, previous methods often idealize the degradation process, and neglect the impact of medium noise and object motion on the distribution of image features, limiting the generalization and adaptability of the model. Previous methods use the reference gradient that is constructed from original images and synthetic ground-truth images. This may cause the network performance to be influenced by some low-quality training data. Our approach utilizes predicted images to dynamically update pseudo-labels, adding a dynamic gradient to optimize the network's gradient space. This process improves image quality and avoids local optima. Moreover, we propose a Feature Restoration and Reconstruction module (FRR) based on a Channel Combination Inference (CCI) strategy and a Frequency Domain Smoothing module (FRS). These modules decouple other degradation features while reducing the impact of various types of noise on network performance. Experiments on multiple public datasets demonstrate the superiority of our method over existing state-of-the-art approaches, especially in achieving performance milestones: PSNR of 25.6dB and SSIM of 0.93 on the UIEB dataset. Its efficiency in terms of parameter size and inference time further attests to its broad practicality. The code will be made publicly available.
翻译:水下图像增强(UIE)是一项具有挑战性的任务,因为水下环境造成的复杂退化。为解决此问题,以往方法常理想化退化过程,忽略介质噪声和物体运动对图像特征分布的影响,限制了模型的泛化能力和适应性。先前方法使用由原始图像和合成真实图像构建的参考梯度,这可能导致网络性能受部分低质量训练数据影响。本文方法利用预测图像动态更新伪标签,通过添加动态梯度优化网络的梯度空间,该过程可提升图像质量并避免陷入局部最优。此外,我们提出基于通道组合推理(CCI)策略的特征恢复与重建模块(FRR)和频域平滑模块(FRS)。这些模块在解耦其他退化特征的同时,降低各类噪声对网络性能的影响。在多个公开数据集上的实验表明,我们的方法优于现有最先进方法,特别是在UIEB数据集上实现了PSNR 25.6dB和SSIM 0.93的性能里程碑。其参数规模与推理时间的效率进一步证实了广泛的实用性。相关代码将公开发布。