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

说明书

本申请涉及图像处理领域，利用对抗网络训练图像生成模型。具体公开了图像生成模型训练方法、图像生成方法、装置、计算机设备及存储介质，该方法包括：获取第一图像集和第二图像集，第一图像集包括多张拍摄图像，第二图像集包括多张漫画图像；根据预设漫画生成算法对拍摄图像进行预处理以得到对应的目标漫画图像；将目标漫画图像作为生成式对抗网络中生成网络的输入，以及将生成网络输出的图像和漫画图像作为生成式对抗网络中判别网络的输入，对生成网络和判别网络进行交替迭代训练，直至判别网络输出的判别概率值大于预设值，将训练后的生成网络将作为图像生成模型。该方法可以训练出将拍摄图像转换成具有漫画风格的图像的模型。

技术领域

本申请涉及图像处理技术领域，尤其涉及一种图像生成模型训练方法、图像生成方法、装置、计算机设备及存储介质。

背景技术

漫画是我们日常生活中广泛使用的艺术形式，具有广泛的应用范围，比如常用于儿童故事教育等，像其他形式的艺术品一样，许多漫画形象都是基于现实世界的场景创作的。然而，将现实世界拍摄的图像转换为漫画风格的图像，这在计算机视觉和计算机图形学中都极具有挑战性，因为漫画风格的图像特征和拍摄图像的图像特征往往差别较大，其中图像特征可例如为人物的发型、衣着、表情、五官等。正是由于两者的差异性过高，将拍摄的图像转换为漫画风格的图像所需要处理的数据维度将会是巨大的，同时所需的图像生成模型也非常难于训练且较为耗时。

发明内容

本申请提供了一种图像生成模型训练方法、图像生成方法、装置、计算机设备及存储介质，以训练出可以将拍摄的图像转换成具有漫画风格的图像的模型，同时，提高训练模型的效率。

第一方面，本申请提供了一种图像生成模型训练方法，其包括：

获取第一图像集和第二图像集，所述第一图像集包括多张拍摄图像，所述第二图像集包括多张漫画图像；

根据预设漫画生成算法对所述拍摄图像进行预处理以得到所述拍摄图像对应的目标漫画图像；

获取预设的生成式对抗网络，所述生成式对抗网络包括生成网络和判别网络；

将所述目标漫画图像作为所述生成网络的输入以及将所述生成网络输出的图像和所述漫画图像作为所述判别网络的输入，对所述生成网络和判别网络进行交替迭代训练；

当所述判别网络输出的判别概率值大于预设值时,保存训练后的生成网络作为图像生成模型，所述图像生成模型用于生成具有漫画风格的图像。

第二方面，本申请还提供了一种图像生成方法，其包括：

获取待处理图像，所述待处理图像为拍摄图像；

根据均值漂移算法对所述待处理图像进行图像分割处理以得到具有层级结构的层级图像；

根据基于流的高斯差分滤波器算法对所述待处理图像进行处理以生成具有边缘轮廓线的边缘图像；

将所述层级图像和所述边缘图像进行图像合成以得到目标图像；

将所述目标图像输入至图像生成模型以生成对应的漫画图像，其中，所述图像生成模型为采用上述的图像生成模型训练方法训练得到的模型。

第三方面，本申请还提供了一种图像生成模型训练装置，其包括：

数据获取单元，用于获取第一图像集和第二图像集，所述第一图像集包括多张拍摄图像，所述第二图像集包括多张漫画图像；

预处理单元，用于根据预设漫画生成算法对所述拍摄图像进行预处理以得到所述拍摄图像对应的目标漫画图像；

网络获取单元，用于获取预设的生成式对抗网络，所述生成式对抗网络包括生成网络和判别网络；