[发明专利]图像处理方法、装置、电子设备和计算机可读存储介质

说明书

本发明实施例公开了一种图像处理方法、装置、电子设备和计算机可读存储介质；本发明实施例在获取图像样本集合后，采用预设图像处理模型对风格图像样本进行风格特征提取，得到风格图像样本的初始风格特征，然后，对初始风格特征进行聚类，以得到图像样本集合对应的风格特征池，根据风格特征池，对初始风格特征进行调整，得到风格图像样本的风格特征，然后，基于风格图像样本的风格特征和原始图像样本对预设图像处理模型进行收敛，得到训练后图像处理模型，该训练后图像处理模型用于根据风格图像对待处理原始图像进行风格转换；该方案通可以提升图像处理的准确性。

技术领域

本发明涉及通信技术领域，具体涉及一种图像处理方法、装置、电子设备和计算机可读存储介质。

背景技术

近年来，随着互联网技术的飞速发展，图像的类型和风格也越来越多。对图像处理的方式也越来越多了，比如，将拍摄的真实照片转换为具有艺术风格的风格图像。现有的图像处理方法往往是直接原始图像进行图像转换或风格迁移等方式生成具有特定艺术风格的风格图像。

在对现有技术的研究和实践过程中，本发明的发明人发现图像转换的方式无法生成不同风格的图像，而风格迁移的方式只是将风格建模为纹理从而实现迁移，对于一些没有纹理信息的艺术风格图像，往往无法保证风格迁移的准确，因此，导致图像处理的准确性降低。

发明内容

本发明实施例提供一种图像处理方法、装置、电子设备和计算机可读存储介质，可以提高图像处理的准确性。

一种图像处理方法，包括：

获取图像样本集合，所述图像样本集合包括原始图像样本和风格图像样本；

采用预设图像处理模型对所述风格图像样本进行风格特征提取，得到所述风格图像样本的初始风格特征；

对所述初始风格特征进行聚类，以得到所述图像样本集合对应的风格特征池；

根据所述风格特征池，对所述初始风格特征进行调整，得到所述风格图像样本的风格特征；

基于所述风格图像样本的风格特征和原始图像样本对所述预设图像处理模型进行收敛，得到训练后图像处理模型，所述训练后图像处理模型用于根据风格图像对待处理原始图像进行风格转换。

相应的，本发明实施例提供一种图像处理装置，包括：

获取单元，用于获取图像样本集合，所述图像样本集合包括原始图像样本和风格图像样本；

提取单元，用于采用预设图像处理模型对所述风格图像样本进行风格特征提取，得到所述风格图像样本的初始风格特征；

聚类单元，用于对所述初始风格特征进行聚类，以得到所述图像样本集合对应的风格特征池；

调整单元，用于根据所述风格特征池，对所述初始风格特征进行调整，得到所述风格图像样本的风格特征；

收敛单元，用于基于所述风格图像样本的风格特征和原始图像样本对所述预设图像处理模型进行收敛，得到训练后图像处理模型，所述训练后图像处理模型用于根据风格图像对待处理原始图像进行风格转换。

可选的，在一些实施例中，所述提取单元，具体可以用于对所述风格图像样本进行数据增广，得到所述风格图像样本对应的多个同源风格图像样本；采用预设图像处理模型的风格特征提取网络对所述同源风格图像样本进行风格特征提取，得到所述风格图像样本的初始风格特征。

可选的，在一些实施例中，所述聚类单元，具体可以用于对所述初始风格特征进行聚类，得到多个风格特征簇；分别计算每一风格特征簇中的风格特征均值，得到每一所述风格特征簇对应的基础风格特征；将所述基础风格特征进行融合，得到所述图像样本集合对应的风格特征池。