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

权利要求书

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

提取待处理图像；将所述待处理图像输入至训练好的图像风格迁移模型中，进行基于目标风格的图像风格迁移处理，得到目标图像，其中，所述目标图像的风格为目标风格；

将所述待处理图像输入至训练好的图像风格迁移模型中，进行基于目标风格的图像风格迁移处理，得到目标图像，包括：分别对所述待处理图像作全局最大池处理和全局平均池处理，得到第一图像；对所述第一图像作卷积处理得到第二图像；对所述第二图像进行通道编码，得到所述目标图像；

所述图像处理方法还包括：获取源样本图像和参考样本图像，所述参考样本图像的风格为所述目标风格；将所述源样本图像输入至第一生成器中，将所述源样本图像的风格转化为所述目标风格，获得目标样本图像；将所述目标样本图像和所述参考样本图像，分别输入至第一判别器中，获得第一判别结果；根据所述第一判别结果，调整所述第一生成器、所述第一判别器中的训练参数，直至所述第一判别器中的判别结果满足预设条件，停止调整；根据调整后的第一生成器，获得所述训练好的图像风格迁移模型；

所述第一生成器包括：第一全局最大池化层和第一全局平均池化层、第一辅助分类器、第一卷积层、多个第一全连接层、第一解码器；

将源样本图像输入至第一生成器中，将所述源样本图像的风格转化为目标风格，获得目标样本图像，包括：

将所述源样本图像输入至第一全局最大池化层，基于第一权重向量对所述源样本图像进行处理，获得第一特征图像；将所述源样本图像输入至所述第一全局平均池化层，基于第二权重向量对所述源样本图像进行处理，获得第二特征图像，将所述第一权重向量和所述第二权重向量输入至所述第一辅助分类器，获得第一分类结果；将所述第一特征图像和所述第二特征图像输入至所述第一卷积层，获得的第三特征图像；将所述第三特征图像输入至所述多个第一全连接层，获得多个参数向量；将所述第三特征图像输入至所述第一解码器，根据所述多个参数向量对所述第三特征图像进行通道编码，获得所述目标样本图像；其中，所述第一权重向量、所述第二权重向量用于调整所述源样本图像中各特征的比重。

2.根据权利要求1所述的方法，其特征在于，所述第一图像包括：全局最大池化处理后的图像和全局平均池化处理后的图像；

所述图像风格迁移模型包括全局最大池化层和全局平均池化层、卷积层、编码层；

所述全局最大池化池，用于对所述待处理图像做全局最大池处理，获得全局最大池化处理后的图像；

所述全局平均池化池，用于对所述待处理图像做全局平均池处理，获得全局平均池化处理后的图像；

所述卷积层，用于对所述第一图像做卷积处理得到所述第二图像；

所述编码层，用于对所述第二图像进行编码，获得所述目标图像。

3.根据权利要求1所述的方法，其特征在于，所述根据所述第一判别结果，调整所述第一生成器、所述第一判别器中的训练参数，直至所述第一判别器中的判别结果满足预设条件，停止调整，包括：

根据所述第一判别结果、所述第一分类结果，调整所述第一生成器、所述第一判别器中的训练参数，获得调整后的第一生成器、调整后的第一判别器；

根据所述调整后的第一生成器，将所述源样本图像的风格转化为目标风格，获得第一目标样本图像；

将所述第一目标样本图像和所述参考样本图像分别输入所述调整后的第一判别器，输出第二判别结果，所述第二判别结果包括：所述第一目标样本图像对应的第一损失值；

当确定所述第一损失值小于预设值时，确定所述第一判别器中判别结果满足预设条件，停止调整。