[发明专利]一种图像修复方法、装置、电子设备及存储介质

权利要求书

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

获取待修复图像；

将所述待修复图像输入去噪网络，得到初始图像；

对所述初始图像进行人脸检测，得到第一检测信息；

在所述第一检测信息满足修复条件时，对所述初始图像进行人脸修复，得到第一修复图像；

将所述第一修复图像输入超分网络，得到所述待修复图像对应的修复图像。

2.根据权利要求1所述的图像修复方法，其特征在于，在所述获取待修复图像之后，在所述将所述待修复图像输入去噪网络之前，所述方法还包括：

对所述待修复图像进行划痕检测；

若检测到所述待修复图像中存在划痕，对所述待修复图像进行划痕强化处理，得到所述待修复图像对应的强化划痕特征图像；

将所述强化划痕特征图像以及所述待修复图像输入划痕修复网络，得到所述待修复图像对应的第二修复图像；

用所述第二修复图像更新所述待修复图像。

3.根据权利要求2所述的图像修复方法，其特征在于，若检测到所述待修复图像中不存在划痕，将所述待修复图像作为所述第二修复图像。

4.根据权利要求1所述的图像修复方法，其特征在于，所述第一检测信息包括以下至少一项：人脸框大小以及人脸偏转角度；所述第一检测信息满足修复条件，包括：

所述人脸框大小大于第一阈值且所述人脸偏转角度小于第二阈值；

所述第一检测信息不满足所述修复条件，包括：

所述人脸框大小小于或等于所述第一阈值，或，所述人脸偏转角度大于或等于所述第二阈值；

其中，所述第一阈值为人脸框尺寸阈值，所述第二阈值为人脸偏转角度阈值。

5.根据权利要求1所述的图像修复方法，其特征在于，在得到第一检测信息之后，所述方法还包括：

在所述第一检测信息不满足所述修复条件时，将所述初始图像作为所述第一修复图像；

将所述第一修复图像输入所述超分网络，得到所述待修复图像对应的所述修复图像。

6.根据权利要求1所述的图像修复方法，其特征在于，在得到所述待修复图像对应的修复图像之后，所述方法还包括：

将所述修复图像输入目标检测网络，获取所述修复图像中的实例图像；

将所述实例图像输入实例着色网络，以提取所述实例图像的实例图像特征，得到实例特征图；

将所述修复图像输入全图像着色网络，以提取所述修复图像的全局图像特征，得到全局特征图；

将所述实例特征图与所述全局特征图进行融合，得到所述修复图像对应的着色图像。

7.根据权利要求6所述的图像修复方法，其特征在于，在得到所述修复图像对应的着色图像之后，所述方法还包括：

获取用户输入的颜色偏好信息；

基于所述颜色偏好信息，获取所述着色图像的颜色信息与所述颜色偏好信息的颜色匹配度；

若所述颜色匹配度大于第三阈值，以所述着色图像更新所述修复图像。

8.根据权利要求1所述的图像修复方法，其特征在于，所述去噪网络包括第一分支和第二分支，将所述待修复图像输入去噪网络，以去除所述待修复图像中的噪声，得到初始图像，包括：

将所述待修复图像输入所述第一分支，基于所述第一分支获取第一去噪图像，所述第一去噪图像为含有盲点的去噪图像；

将所述待修复图像输入所述第二分支，基于所述第二分支获取第二去噪图像，所述第二去噪图像为不含盲点的去噪图像；

将所述第一去噪图像与所述第二去噪图像进行组合，得到所述初始图像。