[发明专利]图像转换模型训练方法、装置、终端设备及存储介质

权利要求书

1.一种图像转换模型训练方法，其特征在于，包括：

获取原始图像和与所述原始图像对应的变换图像，将所述原始图像输入图像转换模型中的第一生成器进行图像转换，并将所述第一生成器的图像转换结果输入所述图像转换模型中的第二生成器进行图像转换，得到第一转换图像；

将所述变换图像输入所述第二生成器进行图像转换，将所述第二生成器的图像转换结果输入所述第一生成器进行图像转换，得到第二转换图像，并将所述第二转换图像与所述原始图像进行图像比对、将所述第一转换图像与所述变换图像进行图像比对，得到图像比对结果；

根据所述图像比对结果对所述第一生成器和所述第二生成器进行参数优化，并将所述原始图像输入参数优化后的所述第一生成器，得到第三转换图像，以及，将所述变换图像对应输入参数优化后的所述第二生成器，得到第四转换图像；

将所述第三转换图像和所述第四转换图像输入所述图像转换模型进行图像检测，得到图像检测结果，所述图像检测结果用于分别指示所述第三转换图像和所述第四转换图像是否为真实图像；

根据所述图像检测结果对所述图像转换模型进行参数更新，直至所述图像转换模型收敛。

2.根据权利要求1所述的图像转换模型训练方法，其特征在于，所述将所述第二转换图像与所述原始图像进行图像比对、将所述第一转换图像与所述变换图像进行图像比对，包括：

分别提取所述第二转换图像、所述原始图像、所述第一转换图像和所述变换图像中的图像轮廓，得到第一轮廓图像、第二轮廓图像、第三轮廓图像和第四轮廓图像；

根据预设轮廓宽度分别对所述第一轮廓图像、所述第二轮廓图像、所述第三轮廓图像和所述第四轮廓图像进行筛选；

将筛选后的所述第一轮廓图像与所述第二轮廓图像进行轮廓比对，得到第一轮廓相似度，并将筛选后的所述第三轮廓图像与所述第四轮廓图像进行轮廓比对，得到第二轮廓相似度。

3.根据权利要求2所述的图像转换模型训练方法，其特征在于，所述将筛选后的所述第一轮廓图像与所述第二轮廓图像进行轮廓比对，得到第一轮廓相似度，并将筛选后的所述第三轮廓图像与所述第四轮廓图像进行轮廓比对，得到第二轮廓相似度，包括：

对筛选后的所述第一轮廓图像和所述第二轮廓图像分别进行多边形逼近处理，得到第一多边形图像和第二多边形图像；

将所述第一多边形图像与所述第二多边形图像进行轮廓比对，得到所述第一轮廓相似度；

对筛选后的所述第三轮廓图像和所述第四轮廓图像分别进行多边形逼近处理，得到第三多边形图像和第四多边形图像；

将所述第三多边形图像与所述第四多边形图像进行轮廓比对，得到所述第二轮廓相似度。

4.根据权利要求1所述的图像转换模型训练方法，其特征在于，所述将所述第三转换图像和所述第四转换图像输入所述图像转换模型进行图像检测，包括：

将所述第三转换图像输入所述图像转换模型中的第一判别器，并指示所述第一判别器根据所述变换图像对所述第三转换图像进行图像检测，得到第一图像检测结果；

将所述第四转换图像输入所述图像转换模型中的第二判别器，并指示所述第二判别器根据所述原始图像对所述第四转换图像进行图像检测，得到第二图像检测结果。

5.根据权利要求4所述的图像转换模型训练方法，其特征在于，所述根据所述图像检测结果对所述图像转换模型进行参数更新，包括：

根据所述第一图像检测结果和所述第二图像检测结果分别进行损失计算，得到第一模型损失值和第二模型损失值；

计算所述第一模型损失值和所述第二模型损失值之间的和，得到模型总损失，并根据所述模型总损失分别对所述第一生成器、所述第二生成器、所述第一判别器和所述第二判别器中的参数进行权值更新。