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

说明书

本申请实施例提供一种图像修复方法、装置、电子设备及存储介质，涉及图像处理技术领域，包括：获取待修复图像；将所述待修复图像输入去噪网络，以去除所述待修复图像中的噪声，得到初始图像；对所述初始图像进行人脸检测，得到第一检测信息，所述第一检测信息包括以下至少一项：人脸框大小以及人脸偏转角度；在所述第一检测信息满足修复条件时，对所述初始图像进行人脸修复，得到第一修复图像；将所述第一修复图像输入超分网络，以提升所述第一修复图像的分辨率，得到所述待修复图像对应的修复图像。

技术领域

本申请实施例涉及图像处理技术领域，具体而言，涉及一种图像修复方法、装置、电子设备及存储介质。

背景技术

随着时间的流逝，对于一些年代较为久远的图像，因为当时的图像采集设备的条件限制使得采集的图像存在瑕疵，或者由于保存不当，使图像出现了破损、模糊、褪色等问题，但是作为珍贵记忆的见证者，人们非常需要将这些图像重新变得清晰。

然而，老旧图像由于可能存在多种不同的问题，例如划痕、模糊、褪色等，在对老照片进行修复时，就需要针对不同类型的图像问题来针对性选择相应的修复方法，这直接导致无法建立标准修复流程适应性的对具有不同问题的图像进行修复，这使得图像修复过程需要投入更多的人力。因此，如何实现对图像修复的流程化处理，成为本领域技术人员亟待解决的技术问题。

发明内容

本申请实施例在于提供一种图像修复方法、装置、电子设备及存储介质，旨在解决如何实现对图像修复的流程化处理的问题。

本申请实施例第一方面提供一种图像修复方法，包括：

获取待修复图像；

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

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

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

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

在一种可选的实施方式中，在所述获取待修复图像之后，在所述将所述待修复图像输入去噪网络之前，所述方法还包括：

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

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

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

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

在一种可选的实施方式中，若检测到所述待修复图像中不存在划痕，将所述待修复图像作为所述第二修复图像。

在一种可选的实施方式中，所述第一检测信息包括以下至少一项：人脸框大小以及人脸偏转角度；所述第一检测信息满足修复条件，包括：

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

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

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

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

在一种可选的实施方式中，在得到第一检测信息之后，所述方法还包括：

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

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