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

说明书

本申请适用于人工智能技术领域，提供了一种图像转换模型训练方法、装置、终端设备及存储介质，该方法包括：将原始图像和变换图像输入第一生成器、第二生成器进行图像转换得到第一转换图像和第二转换图像，将第二转换图像与原始图像、第一转换图像与变换图像进行图像比对；根据图像比对结果对第一生成器和第二生成器进行参数优化，将原始图像、变换图像对应输入参数优化后的第一生成器、第二生成器得到第三转换图像和第四转换图像；对第三转换图像和第四转换图像进行图像检测；根据图像检测结果对图像转换模型进行参数更新，直至图像转换模型收敛。本申请通过对第一生成器和第二生成器进行参数优化，提高了图像转换模型中生成器的准确性。

技术领域

本申请涉及图像处理领域，尤其涉及一种图像转换模型训练方法、装置、终端设备及存储介质。

背景技术

随着人工智能教育的发展，现有的人工智能学习需要非常专业认识的指导，不同的人工智能技术，需要不同的专业人员进行相应的教学，将人工智能技术放入教育，需要考虑简化原本的人工智能技术实现过程。现有的人工智能教育，教学对象一般是学生，如何快速准确的将人像图像转换为卡通图像，显得尤为重要。

现有的人像图像与卡通图像之间均是基于图像转换模型进行图像转换，但现有的图像转换模型训练过程中，均是基于图像中的二维频谱特征对图像转换模型进行训练，进而导致训练后的图像转换模型对图像中的三维特征转换不准确，降低了图像转换模型的准确性。

发明内容

有鉴于此，本申请实施例提供了一种图像转换模型训练方法、装置、终端设备及存储介质，以解决现有技术的图像转换模型训练过程中，由于基于二维频谱特征对图像转换模型进行训练，所导致的图像转换模型准确性差的问题。

本申请实施例的第一方面提供了一种图像转换模型训练方法，包括：

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

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

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

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

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

进一步地，所述将所述第二转换图像与所述原始图像进行图像比对、将所述第一转换图像与所述变换图像进行图像比对，包括：

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

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

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

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