本申请涉及图像处理领域,公开了一种基于transformer网络的水下图像增强方法、系统、设备及介质,方法包括构建增强网络并训练,获取浅层特征图像并馈入第N‑1层编码模块,获得深层特征图像;将第N‑1层编码模块的输出图像与第N层解码模块的输出图像通过第N‑1层特征融合模块融合,输入第N‑1层解码模块后输出残差图像;特征融合模块用于获取对应层编码模块输出图像的概率图,将概率图与上一层解码模块的输出图像进行初步融合,将第N‑1层编码模块输出图像与融合结果进行通道维度的融合;将残差图像与输入图像逐元素相加,获得增强后的水下图像。本申请可以有效解决水下图像存在的颜色偏差、对比度差、复杂场景下增强效果有限等问题。
背景技术
水下图像增强技术作为一种图像处理手段,具有在水下探测和观测领域广泛应用的潜力。水下图像受到水质、光线衰减等因素影响,常常存在模糊、噪声等问题,限制了水下目标的识别和分析。水下图像增强技术可以通过增强对比度、减少噪声等方式改善水下图像质量,从而提高目标识别的准确性和效率;同时,水下图像增强技术能够应用于海洋资源勘探、水下考古、水下搜救等领域,为水下探测工作提供有力支持。
近年来,深度学习技术的快速发展为图像识别和增强提供了强大的工具。然而,传统的深度学习方法在水下图像增强中存在一定的局限性。现有的图像增强方法在水下图像增强方面颜色偏差大、对比度差和复杂场景下增强效果差等限制问题。
实现思路