[发明专利]图像恢复方法、模型训练方法、设备、介质及程序产品

说明书

本申请提供了一种图像恢复方法、模型训练方法、设备、介质及程序产品，涉及图像处理技术领域，旨在对破损图像进行恢复。所述方法包括：获得包含破损区域的破损图像的特征图，并获得所述破损图像关于所述破损区域的权重矩阵，所述权重矩阵中的元素值表征：所述破损图像的各个像素点对所述破损区域的恢复所起的决定作用；基于所述破损图像的特征图和所述权重矩阵，得到偏移矩阵，所述偏移矩阵中的元素值表征：恢复所述破损区域的各个像素点所需的卷积感受野偏移量；基于所述偏移矩阵，对所述破损图像的特征图进行卷积感受野偏移后的卷积操作，得到恢复所述破损区域后的图像的特征图。

技术领域

本申请涉及图像处理技术领域，特别是涉及一种图像恢复方法、模型训练方法、设备、介质及程序产品。

背景技术

在图像处理技术领域，对于一些包含破损区域的破损图像，通常需要先对破损区域进行恢复以得到完整图像，然后对恢复后得到的完整图像进行后续处理，因此，破损图像的恢复效果对后续处理具有重要作用。

然而相关技术中对破损图像进行恢复的方法，往往具有效果不佳的缺点。因此，急需一种恢复效果好的图像恢复方法。

发明内容

鉴于上述问题，本发明实施例提供了一种图像恢复方法、模型训练方法、设备、介质及程序产品，以便克服上述问题或者至少部分地解决上述问题。

本发明实施例的第一方面，提供了一种图像恢复方法，所述方法包括：

获得包含破损区域的破损图像的特征图，并获得所述破损图像关于所述破损区域的权重矩阵，所述权重矩阵中的元素值表征：所述破损图像的各个像素点对所述破损区域的恢复所起的决定作用；

基于所述破损图像的特征图和所述权重矩阵，得到偏移矩阵，所述偏移矩阵中的元素值表征：恢复所述破损区域的各个像素点所需的卷积感受野偏移量；

基于所述偏移矩阵，对所述破损图像的特征图进行卷积感受野偏移后的卷积操作，得到恢复所述破损区域后的图像的特征图。

可选地，所述方法还包括：

利用所述权重矩阵，对所述破损图像的特征图进行处理，得到所述破损图像的加权特征图；

基于所述偏移矩阵，对所述破损图像的特征图进行卷积感受野偏移后的卷积操作，得到恢复所述破损区域后的破损图像的特征图，包括：

基于所述偏移矩阵，对所述破损图像的加权特征图进行卷积感受野偏移后的卷积操作，得到恢复所述破损区域后的破损图像的特征图。

可选地，获得所述破损图像关于所述破损区域的权重矩阵之后，还包括：

利用激活函数对所述权重矩阵进行激活，得到激活后的权重矩阵；

基于所述破损图像的特征图和所述权重矩阵，得到偏移矩阵，包括：

基于所述破损图像的特征图和所述激活后的权重矩阵，得到所述偏移矩阵；和/或

利用所述权重矩阵，对所述破损图像的特征图进行处理，得到所述破损图像的加权特征图，包括：

利用所述激活后的权重矩阵，对所述破损图像的特征图进行处理，得到所述破损图像的加权特征图。

可选地，获得所述破损图像关于所述破损区域的权重矩阵，包括：

获得所述破损图像对应的完整图像；

根据所述完整图像中对应于所述破损区域的图像区域与剩余图像区域的关联，得到所述破损图像关于所述破损区域的权重信息；

对所述权重信息进行特征提取，得到所述权重矩阵。

可选地，基于所述破损图像的特征图和所述权重矩阵，得到偏移矩阵，包括：