[发明专利]一种基于深度学习的水下图像恢复方法

说明书

本发明公开了一种基于深度学习的水下图像恢复方法，属于深度学习和图像处理领域。本发明是基于水下物理模型的水下图像复原方法，创新性地通过将水下物理模型与深度学习相结合的方式，实现对水下退化图像的结构恢复和颜色校正。该发明首先使用暗通道先验模型获取水下退化图像的水下散射图和水下透射率图，然后通过结构恢复网络实现对水下退化图像的物理复原，最后使用颜色校正网络完成对水下退化图像的最终复原。本发明所述方法可以实现对水下退化图像的结构恢复和颜色校正，对在低照度条件下拍摄的水下图像具有很好的适应能力，解决了水下退化图像的结构退化、颜色失真等问题。

技术领域

本发明属于深度学习和图像处理领域，具体涉及一种基于深度学习的水下图像恢复方法。

背景技术

水下图像处理在水下作业项目中是一个热门的话题，也是水下机器人发展的基础。水下图像受到水对光的吸收以及水中悬浮粒子对光的散射(前向散射与后向散射)，导致所捕获的图像可见度低、模糊、对比度下降以及产生色偏，这对水下作业任务带来了严峻的挑战。

在现行的技术中，传统的图像处理算法难以恢复出图像的边缘信息和颜色信息。此外，面对复杂的水下环境，传统水下图像复原方法不具备普适性和迁移性，且受不均匀光照影响较大。另一方面，利用深度学习技术进行图像增强的方法往往不能很好地从机理上去解决水下图像模糊不清、边缘对比度信息丢失等问题。

因此，需要一种结合深度学习技术和水下物理成像模型的方法，从物理上实现对水下图像的恢复，解决水下图像可见度低、色偏、对比度低、模糊等诸多问题，且增加网络模型的泛化性和实用性。

发明内容

本发明提供了一种基于深度学习的水下图像恢复方法，结合了深度学习技术与水下物理成像模型，目的是解决现有技术方法的普适性低、受光照的影响较大以及恢复精确度低等问题。本发明提供的基于深度学的习水下图像恢复方法能够结合水下场景的物理特性，提升了图像恢复的视觉质量。

本发明所提出的技术问题是这样解决的：

一种基于深度学习的水下图像恢复方法，包括以下步骤：

步骤1，将水下退化图像I(z)输入到(Dark Channel Prior，DCP)模型中，得到水下退化图像的水下透射率图t(z)和水下散射图B(z)，并将其标注为目标图像作为辅助训练标签；

步骤2，结构恢复网络(Structure Restoration Network，SRN)训练：将水下退化图像I(z)分别输入到残差密集网络(Residual Dense Network，RDN)和U-Net网络中，且分别以步骤1得出的水下透射率图t(z)和水下散射图B(z)为目标图像进行训练得到SRN模型和权重；

步骤3，颜色校正网络(Color Correction Network，CCN)训练：将步骤2训练好的网络模型分别预测出水下透射率图水下散射图通过水下物理成像模型，即反演得到水下结构恢复图像J(z)，再将J(z)输入到CCN网络模型中，并进行训练得到CCN模型和权重；

步骤4，依次将步骤2、步骤3训练好的网络模型拼接成一个网络模型，再将测试图像输入到拼接好的网络模型中，最终直接输出水下复原图像

本发明步骤2包括：

步骤2-1，利用步骤1中的暗通道先验模型得出水下退化图像I(z)的先验图像，即暗通道图D(x)、水下散射图B(z)和水下透射率图t(z)：

(1)暗通道图D(x)

其中，x表示图像的每个像素点坐标，Ω(x)表示以x为中心的一块领域区域，c表示图像的R、G、B三个通道；

(2)水下散色图B(z)