[发明专利]图像补全方法及装置

说明书

本公开是关于图像补全方法及装置。该方法包括：根据待修复图像，生成n层高斯金字塔，所述n是正整数；通过块匹配方法，补全所述n层高斯金字塔中各个层的图像，得到补全后的所述各个层真实图；根据所述n层高斯金字塔的第n层图像和所述各个层真实图，生成对抗网络的生成网络；根据所述第n层图像和所述生成网络，补全所述待修复图像，得到补全后的补全图。该技术方案通过待修复图像的高斯金字塔、高斯金字塔各个层真实图和对抗网络，得到生成网络，从而能够快速生成清晰完整并且高分辨率的图像，因此，即可以提高补全后图像的分辨率，又可以加快生成网络的学习，得到更真实的图像。

技术领域

本公开涉及图像处理领域，尤其涉及图像补全方法及装置。

背景技术

随着时代的发展，图像补全所应用的情况越来越多，例如，图像放大后细节丢失的情况，图像部分损毁的情况，但是目前的方案仅仅通过图像搜索、匹配、滤波进行补全，补全效果不佳。

发明内容

本公开实施例提供图像补全方法及装置。所述技术方案如下：

根据本公开实施例的第一方面，提供一种图像补全方法，包括：

根据待修复图像，生成n层高斯金字塔，所述n是正整数；

通过块匹配方法，补全所述n层高斯金字塔中各个层的图像，得到补全后的所述各个层真实图；

根据所述n层高斯金字塔的第n层图像和所述各个层真实图，生成对抗网络的生成网络；

根据所述第n层图像和所述生成网络，补全所述待修复图像，得到补全后的补全图。

本公开的实施例提供的技术方案可以包括以下有益效果：通过待修复图像的高斯金字塔、高斯金字塔各个层真实图和对抗网络，得到生成网络，从而能够快速生成清晰完整并且高分辨率的图像，因此，即可以提高补全后图像的分辨率，又可以加快生成网络的学习，得到更真实的图像。

在一个实施例中，所述根据所述第n层图像和所述生成网络，补全所述待修复图像，得到补全后的补全图包括：

获取第i生成图，所述i是0至所述n之间的整数；

将所述第i生成图和所述第i层真实图输入所述生成网络，得到第i-1生成图；

其中，所述第0生成图是所述补全图；所述第n生成图是从所述第n层图像中，随机上采样得到与所述第n层图像分辨率相同的图像。

本公开的实施例提供的技术方案可以包括以下有益效果：通过循环的方式，增加图像像素、实现生成网络补全图像。

在一个实施例中，所述根据所述n层高斯金字塔的第n层图像和所述各个层真实图，生成对抗网络的生成网络包括：

通过所述第n层图像和所述各个层真实图，训练出满足目标函数要求的对抗网络的判别网络；

通过所述第n层图像、所述各个层真实图和所述判别网络，训练出满足目标函数要求的所述生成网络；

其中，所述目标函数用于衡量补全过程中的损失。

本公开的实施例提供的技术方案可以包括以下有益效果：通过训练判别网络，从而训练出生成网络，从而实现图像补全。

在一个实施例中，所述通过所述第n层图像和所述各个层真实图，训练出满足目标函数要求的对抗网络的判别网络包括：

获取第j更新图和第j层真实图，所述j是1到所述n之间的整数；

将所述第j层真实图和所述第j更新图输入改进判别网络，得到第一判断结果；

当所述第一判断结果表征所述第j更新图和所述第j层真实图是同一图像时，保持所述改进判别网络的参数；