摘要

水下图像由于波长和距离依赖的衰减和散射而受到色偏和低对比度的影响。为了解决这两个退化问题，我们提出了一种基于介质传输引导的多颜色空间嵌入的水下图像增强网络，称为Ucolor。具体而言，我们首先提出了一种多颜色空间编码器网络，该网络通过将不同颜色空间的特性融入统一结构中，丰富了特征表示的多样性。结合注意力机制，从多个颜色空间中提取出最具区分性的特征被自适应地整合并突出显示。受水下成像物理模型的启发，我们设计了一种介质传输（指示场景辐射到达相机的比例）引导的解码器网络，以增强网络对质量退化区域的响应。因此，我们的网络能够通过利用多颜色空间嵌入以及基于物理模型和学习方法的优势，有效提高水下图像的视觉质量。大量实验表明，我们的Ucolor在视觉质量和定量指标方面均优于现有最先进方法。