[发明专利]一种修复图像的方法、装置以及存储介质

权利要求书

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

根据待修复图像预测目标掩膜图像，所述目标掩膜图像包括与所述待修复图像的像素一一对应的权重，任一所述权重大于或者等于0且小于或者等于1，每个所述权重表征所述待修复图像中相应像素的未受损程度；

根据所述目标掩膜图像对所述待修复图像进行编码，得到所述待修复图像的编码特征图，所述编码特征图的每个特征值均是所述待修复图像的相应特征值与所述目标掩膜图像中对应的权重相乘得到；

对所述编码特征图解码得到所述待修复图像对应的修复图像。

2.根据权利要求1所述的修复图像的方法，其特征在于，所述根据所述目标掩膜图像对所述待修复图像进行编码，得到所述待修复图像的编码特征图，包括：

对所述待修复图像和所述目标掩膜图像进行多级编码，得到所述编码特征图。

3.根据权利要求2所述的修复图像的方法，其特征在于，其中，任一级编码操作包括：

从所述待修复图像的输入数据中提取所述待修复图像的第一初始特征图，所述待修复图像的输入数据是所述待修复图像数据或者上一级编码输出的特征图；

使用本级编码过程对应的降维规则对所述第一初始特征图降维得到第二初始特征图，以及对第一初始掩膜图像降维得到第二初始掩膜图像，所述第二初始特征图的特征值与所述第二初始掩膜图像的权重值一一对应，所述第一初始掩膜图像是所述目标掩膜图像或者所述上一级编码输出的掩膜图像；

将所述第二特征图的每个特征值与所述第二初始掩膜图像的权重值对应相乘，得到所述本级编码对应的特征图。

4.根据权利要求1所述的修复图像的方法，其特征在于，所述根据待修复图像预测目标掩膜图像，包括：

提取所述待修复图像的特征图，所述特征图表征所述待修复图像中每个像素的特征值；

根据所述特征图获得所述每个像素是否是未受损像素的概率分布；

根据所述每个像素对应的概率分布得到所述每个像素对应的权重，各个像素的权重组成所述目标掩膜图像。

5.根据权利要求1至4中任一项所述的修复图像的方法，其特征在于，

根据待修复图像预测目标掩膜图像包括：使用预训练的掩膜预测模型预测所述目标掩膜图像；

在根据所述目标掩膜图像对所述待修复图像进行编码，得到所述待修复图像的编码特征图之前，还包括：将所述待修复图像和所述目标掩膜图像输入预训练的图像修复模型。

6.根据权利要求5所述的修复图像的方法，其特征在于，还包括：

使用预配置的多个训练样本训练第一模型得到所述掩膜预测模型，以及训练第二模型得到所述图像修复模型，所述训练样本实现为受损图像，其中，每个所述训练样本对应的受损图像均由任一二值掩膜图像和一个未受损图像得到。

7.根据权利要求6所述的修复图像的方法，其特征在于，所述使用预配置的多个训练样本训练第一模型得到所述掩膜预测模型包括：

将所述多个训练样本中的任一训练样本输入第一待训练模型，得到所述训练样本对应的预测掩膜图像，所述第一待训练模型根据所述第一模型得到；

计算所述预测掩膜图像与得到所述训练样本的二值掩膜图像的损失函数值；

若所述损失函数值收敛，将所述第一待训练模型确定为所述掩膜预测模型；

若所述损失函数值发散，调整所述第一待训练模型的参数得到新的第一待训练模型，重复执行所述将所述多个训练样本中的任一训练样本输入第一待训练模型的步骤。