针对水下图像的光晕问题,提出双分支网络实现光晕分离与多尺度恢复。
Halo Separation-guided Underwater Multi-scale Image Restoration

- 分两阶段:先分离光晕层,再恢复被遮蔽图像信息
- 在真实人工光源水下图像上测试,显著提升视觉质量
- 引入径向梯度约束,有效抑制光晕并增强恢复效果
水下自主航行器(AUV)拍摄的图像常受人工光源影响,产生前景光晕,严重降低图像质量。现有增强方法未充分考虑此问题,处理人工光源场景时鲁棒性差。为此,本文设计一种基于迭代结构的单张光晕图像校正方法。网络分为两个子网:光晕层分离子网通过梯度最小化分离光晕;多尺度恢复子网则恢复被光晕遮蔽的信息。使用UIEB和EUVP合成数据集训练,确保模型学习到水下光晕特征。同时收集大量真实人工光源下的水下图像进行测试。分析了水下光晕图像的亮度分布特性,引入径向梯度约束以更有效地消除光晕,提升图像恢复效果。
原文摘要 · Abstract (English)
Underwater images captured by Autonomous Underwater Vehicles (AUVs) are inevitably affected by artificial light sources, which often produce halos in the foreground of the camera and seriously interfere with the quality of the image. The existing underwater image enhancement methods fail to fully consider this key problem, and the robustness of processing images under artificial light scenes is poor. In practical applications, since underwater image enhancement itself is a very challenging task, the influence of artificial light sources will lead to serious degradation of image performance and affect subsequent vision tasks. In order to effectively deal with this problem, this paper designs a single halo image correction method based on an iterative structure. The network is mainly divided into two sub-networks, one is the halo layer separation sub-network which aims to separate the halo by gradient minimization, and the other is the multi-scale recovery sub-network which aims to recover the image information masked by halo. The UIEB and EUVP synthetic datasets are used for training to ensure that the network can fully learn the characteristics and laws of underwater halo images. Then a large number of halo images taken in an underwater environment with real artificial light are collected for testing. In addition, the brightness distribution characteristics of underwater halo images are analyzed and the radial gradient is introduced to constraint eliminate halo to improve the effect of underwater image restoration.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。