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

权利要求书

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

根据样本图像对生成器进行训练，获取训练好的生成器；

基于样本图像对编码器进行训练，获取训练好的编码器；

基于风格样本图像对所述训练好的生成器进行训练得到风格化生成器，并根据所述训练好的生成器、训练好的编码器以及所述风格化生成器获取风格化模型。

2.根据权利要求1所述的模型训练方法，其特征在于，所述根据样本图像对生成器进行训练，获取训练好的生成器，包括：

采用对抗生成方式，基于所述样本图像对所述生成器进行训练，得到训练好的生成器。

3.根据权利要求1所述的模型训练方法，其特征在于，所述基于样本图像对编码器进行训练，得到训练好的编码器，包括：

结合所述训练好的生成器对编码器进行训练，得到训练好的编码器。

4.根据权利要求3所述的模型训练方法，其特征在于，所述结合所述训练好的生成器对编码器进行训练，得到训练好的编码器，包括：

固定所述训练好的生成器的参数，以最小化重建误差为训练目标，对编码器进行训练，获取训练好的编码器。

5.根据权利要求1所述的模型训练方法，其特征在于，所述基于风格样本图像对所述训练好的生成器进行训练得到风格化生成器，包括：

根据所述风格样本图像对训练好的生成器的模型参数进行调整，以得到所述风格化生成器。

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

获取待处理图像；

基于风格化模型对所述待处理图像进行特征提取得到所述待处理图像的特征向量，并根据所述特征向量生成所述待处理图像的风格化图像；所述风格化模型由权利要求1-5任意一项所述的模型训练方法训练得到。

7.根据权利要求6所述的图像处理方法，其特征在于，所述基于风格化模型对所述待处理图像进行特征提取，获取所述待处理图像的特征向量，包括：

基于所述风格化模型对所述待处理图像进行特征提取，获取一个主特征向量以及多个偏移向量；

将所述主特征向量以及所述多个偏移向量进行融合，得到所述特征向量。

8.一种模型训练装置，其特征在于，包括：

生成器训练模块，用于根据样本图像对生成器进行训练，获取训练好的生成器；

编码器训练模块，用于基于样本图像对编码器进行训练，获取训练好的编码器；

风格化生成器训练模块，用于基于风格样本图像对所述训练好的生成器进行训练得到风格化生成器，并根据所述训练好的生成器、训练好的编码器以及所述风格化生成器获取风格化模型。

9.一种图像处理装置，其特征在于，包括：

图像获取模块，用于获取待处理图像；

图像生成模块，用于基于风格化模型对所述待处理图像进行特征提取得到所述待处理图像的特征向量，并根据所述特征向量生成所述待处理图像的风格化图像；所述风格化模型根据权利要求1-5中任意一项所述的模型训练方法训练得到。

10.一种电子设备，其特征在于，包括：

处理器；以及

存储器，用于存储所述处理器的可执行指令；

其中，所述处理器配置为经由执行所述可执行指令来执行权利要求1-5任意一项所述的模型训练方法或权利要求6-7任意一项所述的图像处理方法。

11.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现权利要求1-5任意一项所述的模型训练方法或权利要求6-7任意一项所述的图像处理方法。