[发明专利]图像处理方法、装置、计算机设备和存储介质

权利要求书

1.一种图像处理方法，其特征在于，所述方法包括：

获取原始图像和目标生成尺寸；

按照所述目标生成尺寸的高度和宽度的比值对所述原始图像进行拓展，得到拓展图像，所述拓展图像包括拓展区域和原始图像对应的原始区域；

根据所述拓展图像的拓展区域和原始区域生成掩码图像，所述掩码图像与所述拓展图像尺寸相同；

将所述掩码图像、所述目标生成尺寸和所述原始图像输入图像生成网络，输出生成图像，所述生成图像的尺寸与所述目标生成尺寸相同，所述图像生成网络为目标对抗网络的子网络。

2.根据权利要求1所述的方法，其特征在于，包括：

所述按照所述目标生成尺寸的高度和宽度的比值对所述原始图像进行拓展，得到拓展图像，包括：在所述原始图像中的两侧增加像素值为零的像素点，得到所述拓展图像，以使所述拓展图像的高度与宽度比值与所述目标生成尺寸的高度和宽度的比值相同；

所述根据所述拓展图像的拓展区域和原始区域生成掩码图像，包括：将所述拓展图像中的所述原始区域的区域元素设置为1，得到所述掩码图像。

3.根据权利要求1所述的方法，其特征在于，所述目标对抗网络包括所述图像生成网络和图像判别网络，生成所述目标对抗网络的步骤，包括：

获取多个训练图像对，所述训练图像对包括训练图像和对应的目标生成图像；

输入各个所述训练图像对至初始对抗网络的图像生成网络，输出各个所述训练图像对应的第一生成图像；

根据各个所述训练图像的第一生成图像和对应的目标生成图像的差异度，计算所述初始对抗网络的图像生成网络的第一损失值；

将各个所述训练图像的第一生成图像和目标生成图像输入所述初始对抗网络中的图像判别网络，输出所述训练图像的第一生成图像的识别结果和所述目标生成图像的识别结果；

根据所述第一生成图像的识别结果和所述目标生成图像的识别结果的识别差异度，计算所述初始对抗网络的图像判别网络的第二损失值，根据所述识别差异度计算识别准确率；

当所述第一损失值小于或等于第一预设损失值，且所述识别准确率位于预设准确率区间时，得到所述目标对抗网络。

4.根据权利要求3所述的方法，其特征在于，所述方法包括：

当所述第一损失值大于所述第一预设损失值时，根据所述第一损失值和所述第二损失值更新所述初始对抗网络的网络参数，得到中间对抗网络，输入各个所述训练图像对至所述中间对抗网络，直至当所述第一损失值小于或等于所述第一预设损失值，且所述识别准确率位于所述预设准确率区间时，得到所述目标对抗网络。

5.根据权利要求4所述的方法，其特征在于，所述根据所述第一损失值和所述第二损失值更新所述初始对抗网络的网络参数，得到中间对抗网络，包括：

根据预设回传算法回传所述第二损失值，并更新所述初始对抗网络中的图像判别网络的网络参数，得到中间图像判别模型和对应的回传误差；

根据所述回传误差和所述第一损失值更新所述初始对抗网络中的图像生成网络的网络参数，得到中间图像生成网络，所述中间图像生成网络和所述中间图像判别网络组成所述中间对抗网络。

6.根据权利要求3所述的方法，其特征在于，所述获取多个训练图像对，包括：

获取多个所述目标生成图像，各个所述目标生成图像为横屏图像；

从各个所述目标生成图像中截取部分图像区域，得到各个所述目标生成图像对应的训练图像，所述目标生成图像和所述目标生成图像对应的训练图像组成所述训练图像对。