[发明专利]基于对抗生成网络的图像印章去除方法、装置及相关设备

权利要求书

1.一种基于对抗生成网络的图像印章去除方法，其特征在于，包括：

获取不同图像类别的样本图像以及图像处理模型，其中，所述图像类别包括有印章类别和无印章类别，所述图像处理模型包括第一对抗生成网络、第二对抗生成网络以及分类模型；

利用所述分类模型对样本图像进行图像判别，得到有印章类别的样本图像和无印章类别的样本图像，所述图像判别还用于确定所述第一对抗生成网络和第二对抗生成网络输出的图像为有印章类别或无印章类别；

将有印章类别的样本图像作为第一原始图像先输入至所述第一对抗生成网络进行去印章处理，得到第一生成图像，并将所述第一生成图像输入所述分类模型进行图像判别；再将所述第一生成图像输入至所述第二对抗生成网络进行印章还原处理，得到第一还原图像，并将所述第一还原图像输入所述分类模型进行图像判别；

将无印章类别的样本图像作为第二原始图像先输入至所述第二对抗生成网络进行印章还原处理，得到第二生成图像，并将所述第二生成图像输入所述分类模型进行图像判别；再将所述第二生成图像输入至所述第一对抗生成网络进行去印章处理，得到第二还原图像，并将所述第二还原图像输入所述分类模型进行图像判别；

根据预置的损失函数计算所述第一原始图像和第一还原图像以及第二原始图像和第二还原图像的损失值，根据所述损失值优化所述图像处理模型的模型参数，得到最优的图像处理模型；

获取待去除印章的目标处理图像，将所述目标处理图像输入至所述第一对抗生成网络进行去印章处理，得到所述目标处理图像对应的无印章图像。

2.根据权利要求1所述的图像印章去除方法，其特征在于，所述第一对抗生成网络及第二对抗生成网络均包括生成器网络和判别器网络，所述生成器网络采用Unet网络结构。

3.根据权利要求2所述的图像印章去除方法，其特征在于，所述生成器网络包括三个下采样单元、三个残差模块以及三个上采样单元，每一所述残差模块由两个卷积层构成。

4.根据权利要求1所述的图像印章去除方法，其特征在于，所述将有印章类别的样本图像作为第一原始图像先输入至所述第一对抗生成网络进行去印章处理，得到第一生成图像，包括：

通过所述第一对抗生成网络的下采样单元对所述第一原始图像进行预设多个维度的下采样特征提取，得到所述第一原始图像在所述多个维度内的第一下采样特征；

通过所述第一对抗生成网络的残差模块组合所述多个维度内的第一下采样特征得到第一中间特征，并通过所述第一对抗生成网络的上采样单元对所述第一中间特征进行上采样特征提取，得到第一上采样特征；

根据所述第一上采样特征还原出与所述第一原始图像同一维度的图像，作为所述第一生成图像。

5.根据权利要求1所述的图像印章去除方法，其特征在于，所述将所述第一生成图像输入至所述第二对抗生成网络进行印章还原处理，得到第一还原图像，包括：

通过所述第二对抗生成网络的下采样单元对所述第一生成图像进行多个维度的下采样特征提取，得到所述第一生成图像在所述多个维度内的第二下采样特征；

通过所述第二对抗生成网络的残差模块组合所述多个维度内的第二下采样特征得到第二中间特征，并通过所述第二对抗生成网络的上采样单元对所述第二中间特征进行上采样特征提取，得到第二上采样特征；

根据所述第二上采样特征还原出与所述第一生成图像同一维度的图像，得到所述第一还原图像。

6.根据权利要求1所述的图像印章去除方法，其特征在于，所述损失函数包括第一对抗生成网络和第二对抗生成网络的一致性损失、分类模型损失、第一对抗生成网络的对抗损失和第二对抗生成网络的对抗损失。