[发明专利]一种图像补全方法、装置及电子设备

说明书

本发明实施例提供了一种图像补全方法、装置及电子设备。其中，所述方法包括：确定待补全图像中缺失图像信息的待补全图像区域和细节图像区域；根据所述待补全图像区域与所述细节图像区域的重合程度，确定目标图像补全模型；所述目标图像补全模型包括细节图像补全模型或通用补全模型；所述细节图像补全模型为针对所述预设内容训练得到的图像补全模型；所述通用补全模型为针对所述预设内容以外的内容训练得到的图像补全模型；基于所述目标图像补全模型对所述待补全图像区域缺失的图像信息进行补全，得到补全结果。可以通过分析待补全图像区域中的内容，选择相应的图像补全模型对待补全图像区域所丢失的图像信息进行补全，以提高补全结果的视觉效果。

技术领域

本发明涉及图像处理技术领域，特别是涉及一种图像补全方法、装置及电子设备。

背景技术

出于一些原因，图像中部分像素点的图像信息可能丢失，例如图像中可能添加有水印，水印会遮挡住图片中的部分区域，导致这些区域的像素点的图像信息丢失，影响图像的视觉效果。相关技术中，可以利用深度学习得到的图像补全模型对图像中缺失的图像信息进行补全。

图像补全模型的工作原理为根据缺失图像信息的像素点附近的未缺失的图像信息类推得到这些像素点的图像信息。但是图像中可能存在具有一定细节的部分区域，这部分区域的颜色、纹理和图像中其他区域的颜色、纹理存在较大区别，例如图像中可能存在部分区域用于表示人员，而其他区域表示风景，而人员图像的颜色、纹理和风景图像的颜色、纹理存在较大区别。

因此如果图像补全模型按照和对其他区域相同的类推方式，对该部分区域丢失的图像信息进行补全，所补全的图像信息可能和该部分区域实际所丢失的图像信息相差较大。因此，如果图像中该部分区域的图像信息丢失，则相关技术中利用图像补全模型对该图像进行补全，得到的补全结果与真实图像相差较大，可能导致用户产生违和感，因此视觉效果较差。

发明内容

本发明实施例的目的在于提供一种图像补全方法、装置及电子设备，以实现提高得到的补全结果的视觉效果。具体技术方案如下：

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

确定待补全图像中缺失图像信息的待补全图像区域和细节图像区域，所述细节图像区域为所述待补全图像中用于表示预设内容的图像区域；

根据所述待补全图像区域与所述细节图像区域的重合程度，确定目标图像补全模型；所述目标图像补全模型包括细节图像补全模型或通用补全模型；所述细节图像补全模型为针对所述预设内容训练得到的图像补全模型；所述通用补全模型为针对所述预设内容以外的内容训练得到的图像补全模型；

基于所述目标图像补全模型对所述待补全图像区域缺失的图像信息进行补全，得到补全结果。

在一种可能的实施例中，在所述根据所述待补全图像区域与所述细节图像区域的重合程度，确定目标图像补全模型之前，所述方法还包括：

统计所述待补全图像区域中属于细节图像区域的像素点的数目与所述待补全图像区域中所有像素点的数目的比例，作为所述待补全图像区域与所述细节图像区域的重合程度。

在一种可能的实施例中，所述细节图像补全模型预先通过以下方式训练得到：

获取原始样本图像和缺失样本图像，所述原始样本图像中包括所述预设内容，所述缺失样本图像为所述原始样本图像中用于表示所述预设内容的部分图像信息丢失后的图像；

将所述缺失样本图像输入至预设生成模型，得到所述预设生成模型输出的补全图像；

基于所述补全图像与所述原始样本图像的差异构建第一损失函数；

基于所述第一损失函数，调整所述预设生成模型的模型参数，直至所述第一损失函数收敛，得到所述细节图像补全模型。