[发明专利]缺损图像复原方法、装置及电子设备

说明书

本发明公开了一种缺损图像复原方法、装置及电子设备。其中，该方法包括：获取目标缺损图像；将目标缺损图像输入至图像修复生成器中的编码模块中，得到与目标缺损图像对应的目标特征图，其中，编码模块包括编码卷积层，编码密集网络块，编码卷积层用于提取特征图，编码密集网络块用于扩宽特征图的通道数；将目标特征图输入至图像修复生成器中的解码模块中，复原目标缺损图像，得到目标完整图像，其中，解码模块包括解码密集网络块，解码反卷积层，编码密集网络块用于缩减特征图的通道数，解码反卷积层用于还原特征图。本发明解决了相关技术中进行缺损图像的复原时，难以对任意形状的缺损图像完成任意像素的复原的技术问题。

技术领域

本发明涉及图像处理领域，具体而言，涉及一种缺损图像复原方法、装置及电子设备。

背景技术

图像修复是一种对破损或者存在缺失的图像区域进行补全的技术，它在缺失区域插入所预测的内容，图像修复为目标移除提供了技术支持。尽管已经提出了许多图像修复的方法，但是图像修复仍然是一个具有挑战性的问题，相关技术中进行缺损图像的复原时，还是难以对任意形状的缺损图像完成任意像素的复原。

针对上述的问题，目前尚未提出有效的解决方案。

发明内容

本发明实施例提供了一种缺损图像复原方法、装置及电子设备，以至少解决相关技术中进行缺损图像的复原时，难以对任意形状的缺损图像完成任意像素的复原的技术问题。

根据本发明实施例的一个方面，提供了一种缺损图像复原方法，包括：获取目标缺损图像；将所述目标缺损图像输入至图像修复生成器中的编码模块中，得到与所述目标缺损图像对应的目标特征图，其中，所述编码模块包括编码卷积层，编码密集网络块，所述编码卷积层用于提取特征图，所述编码密集网络块用于扩宽特征图的通道数；将所述目标特征图输入至所述图像修复生成器中的解码模块中，复原所述目标缺损图像，得到目标完整图像，其中，所述解码模块包括解码密集网络块，解码反卷积层，所述编码密集网络块用于缩减特征图的通道数，所述解码反卷积层用于还原特征图。

可选地，所述将所述目标缺损图像输入至图像修复生成器中的编码模块中，得到与所述目标缺损图像对应的目标特征图，包括：在所述编码卷积层包括第一卷积层，第二卷积层，第三卷积层，编码密集网络块包括第一密集网络块，第二密集网络块的情况下，将所述目标缺损图像输入至图像修复生成器中的所述编码模块中，所述目标缺损图像依次经过顺序连接的所述第一卷积层，所述第一密集网络块，所述第二密集网络块，所述第二卷积层与所述第三卷积层，得到与所述目标缺损图像对应的目标特征图，其中，所述编码卷积层中包括的卷积层的卷积核与步长依据所述目标缺损图像的像素值确定。

可选地，所述将所述目标特征图输入至所述图像修复生成器中的解码模块中，复原所述目标缺损图像，得到目标完整图像，包括：在所述解码反卷积层包括第一反卷积层，第二反卷积层，第三反卷积，解码密集网络块包括第三密集网络块，第四密集网络块，第五密集网络块的情况下，将所述目标特征图输入至所述图像修复生成器中的所述解码模块中，所述目标特征图依次经过顺序连接的所述第三密集网络块，所述第一反卷积层，所述第四密集网络块，所述第二反卷积层，所述第五密集网络块与所述第三反卷积层，复原所述目标缺损图像，得到所述目标完整图像，其中，所述编码反卷积层中包括的反卷积层的卷积核与步长依据所述目标特征图的像素值确定。

可选地，所述将所述目标缺损图像输入至图像修复生成器中的编码模块中，得到与所述目标缺损图像对应的目标特征图之前，还包括：获取初始图像修复生成器；采用多组样本数据训练所述初始图像修复生成器，得到所述图像修复生成器，其中，所述多组样本数据包括样本缺损图像，样本完整图像，真实完整图像。