[发明专利]模型训练方法、图像处理方法及装置、电子设备、介质

说明书

本公开实施例是关于一种模型训练方法、图像处理方法及装置、电子设备、存储介质，涉及图像处理技术领域，该模型训练方法包括：根据样本图像对生成器进行训练，获取训练好的生成器；基于样本图像对编码器进行训练，获取训练好的编码器；基于风格样本图像对所述训练好的生成器进行训练得到风格化生成器，并根据所述训练好的生成器、训练好的编码器以及所述风格化生成器获取风格化模型。本公开的技术方案能够提高模型训练效率以及模型的准确性。

技术领域

本公开涉及图像处理技术领域，具体而言，涉及一种模型训练方法、图像处理方法、模型训练装置、图像处理装置、电子设备以及计算机可读存储介质。

背景技术

在图像处理过程中，可以对人脸图像进行风格化处理，以实现多样性处理。

相关技术中，一般采用端到端的风格化模型对人脸图像进行风格化处理，或者是采用3D建模技术创建的人脸风格化三维模型来进行风格化处理。

上述方式中，在训练风格化模型时需要大量的风格化图像数据，但是大量的风格化图像数据难获取，因此模型的准确性较低，导致得到的风格化图像准确性较低。基于3D建模技术的人脸风格化方法的操作步骤复杂，并且鲁棒性较差。

需要说明的是，在上述背景技术部分公开的信息仅用于加强对本公开的背景的理解，因此可以包括不构成对本领域普通技术人员已知的现有技术的信息。

发明内容

本公开的目的在于提供一种模型训练方法、图像处理方法及装置、电子设备、存储介质，进而至少在一定程度上克服由于相关技术的限制和缺陷而导致的模型准确性较低的问题。

本公开的其他特性和优点将通过下面的详细描述变得显然，或部分地通过本公开的实践而习得。

根据本公开的一个方面，提供一种模型训练方法，包括：根据样本图像对生成器进行训练，获取训练好的生成器；基于样本图像对编码器进行训练，获取训练好的编码器；基于风格样本图像对所述训练好的生成器进行训练得到风格化生成器，并根据所述训练好的生成器、训练好的编码器以及所述风格化生成器获取风格化模型。

根据本公开的一个方面，提供一种图像处理方法，包括：获取待处理图像；基于风格化模型对所述待处理图像进行特征提取得到所述待处理图像的特征向量，并根据所述特征向量生成所述待处理图像的风格化图像；所述风格化模型由上述任意一项所述的模型训练方法训练得到。

根据本公开的一个方面，提供一种模型训练装置，包括：生成器训练模块，用于根据样本图像对生成器进行训练，获取训练好的生成器；编码器训练模块，用于基于样本图像对编码器进行训练，获取训练好的编码器；风格化生成器训练模块，用于基于风格样本图像对所述训练好的生成器进行训练得到风格化生成器，并根据所述训练好的生成器、训练好的编码器以及所述风格化生成器获取风格化模型。

根据本公开的一个方面，提供一种图像处理装置，包括：图像获取模块，用于获取待处理图像；图像生成模块，用于基于风格化模型对所述待处理图像进行特征提取得到所述待处理图像的特征向量，并根据所述特征向量生成所述待处理图像的风格化图像；所述风格化模型根据上述任意一项所述的模型训练方法训练得到。

根据本公开的一个方面，提供一种电子设备，包括：处理器；以及存储器，用于存储所述处理器的可执行指令；其中，所述处理器配置为经由执行所述可执行指令来执行上述任意一项所述的模型训练方法或上述任意一项所述的图像处理方法。

根据本公开的一个方面，提供一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时实现上述任意一项所述的模型训练方法或上述任意一项所述的图像处理方法。

根据本公开的一个方面，提供一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时实现上述任意一项所述的模型训练方法或上述任意一项所述的图像处理方法。