[发明专利]图像修复方法、装置和电子设备

权利要求书

1.一种图像修复方法，其特征在于，包括：

获取待修复图像；

生成待修复图像中的噪声区域的修复图像；

将所述噪声区域的修复图像贴合到所述待修复图像上得到修复后的图像；

其中，所述生成待修复图像中的噪声区域的修复图像，包括：将所述待修复图像输入图像修复模型得到所述噪声区域的修复图像；

所述图像修复模型通过以下训练步骤得到：获取训练数据集合，其中所述训练数据集合中的训练数据是由原始图像、在原始图像中添加了噪声的第一图像以及由所述噪声生成的第二图像所组成的三元数据；根据所述原始图像以及所述第二图像生成噪声区域的修复图像；初始化图像修复模型的参数；将所述第一图像输入所述图像修复模型得到预测的噪声区域的修复图像；根据所述噪声区域的修复图像以及所述预测的噪声区域的修复图像计算误差；基于所述误差更新所述图像修复模型的参数；使用更新后的参数迭代上述参数更新的过程直至达到收敛条件；将达到收敛条件时所得到的参数作为训练好的图像修复模型的参数。

2.如权利要求1所述的图像修复方法，其中所述生成待修复图像中的噪声区域的修复图像，包括：

根据所述待修复图像生成噪声区域的轮廓；

根据所述待修复图像生成中间图像；

根据所述噪声区域的轮廓以及中间图像生成所述噪声区域的修复图像。

3.如权利要求2所述的图像修复方法，其中所述修复模型包括编码层、第一解码层和第二解码层，其中：

所述将所述待修复图像输入所述编码层得到所述待修复图像的特征图；

其中，所述根据所述待修复图像生成噪声区域的轮廓，包括：

将所述待修复图像的特征图输入所述第一解码层得到所述噪声区域的轮廓；

所述根据所述待修复图像生成中间图像，包括：

将所述待修复图像的特征图输入所述第二解码层得到所述中间图像。

4.如权利要求1所述的图像修复方法，其中所述获取训练数据集合，包括：

获取原始图像；

在原始图像上随机生成噪声得到第一图像；

根据所述噪声生成仅包含所述噪声的第二图像。

5.如权利要求4所述的图像修复方法，其中所述第一图像中的噪声的像素点与所述噪声图像中的噪声像素点一一对应。

6.如权利要求1-5任一项所述的图像修复方法，其中所述第二图像为背景为黑色、噪声为白色的二值化图像。

7.如权利要求1所述的图像修复方法，其中所述将所述第一图像输入所述图像修复模型得到预测的噪声区域的修复图像，包括：

将所述第一图像输入所述图像修复模型的编码层得到预定尺度大小的第一特征图像；

将所述第一特征图像输入所述图像修复模型的第一解码层得到与所述第一图像大小相同的预测的第二图像；

将所述第一特征图像输入所述图像修复模型的第二解码层得到与所述第一图像大小相同的预测的中间图像；

根据所述预测第二图像和所述预测的中间图像计算得到预测的噪声区域的修复图像。

8.如权利要求7所述的图像修复方法，其中所述解码层包括多个卷积层，所述多个卷积层对所述第一图像进行下采样得到所述第一特征图像。

9.如权利要 求7所述的图像修复方法，其中所述第一解码层和所述第二解码层分别包括多个反卷积层，所述第一解码层的多个反卷积层对所述第一特征图进行上采样得到预测的第二图像；所述第二解码层的多个反卷积层对所述第一特征图进行上采样得到预测的中间图像。

10.如权利要求7所述的图像修复方法，其中所述根据所述预测的第二图像和所述预测的中间图像计算得到预测的噪声区域的修复图像，包括：

将所述的预测的第二图像和所述预测的中间图像进行对应像素值相乘计算得到预测的噪声区域的修复图像。