[发明专利]图像处理方法、装置、电子设备及存储介质

说明书

本申请公开了一种图像处理方法、装置、电子设备及存储介质，该图像处理方法包括：获取第一图像、第二图像以及初始图像；利用预先训练的至少一个特征提取模型，提取第一图像的内容特征、第二图像的风格特征、初始图像的内容特征以及初始图像的风格特征；根据初始图像的内容特征与第一图像的内容特征的差异，确定内容损失函数的值，以及根据初始图像的风格特征中不同特征之间的相似度，与第二图像的风格特征中对应的不同特征之间的相似度的差异，确定风格损失函数的值；根据内容损失函数的值和风格损失函数的值确定总损失函数的值；根据总损失函数的值调整初始图像，直至得到目标图像。本方法可以较好地实现图像风格的迁移。

技术领域

本申请涉及定位技术领域，更具体地，涉及一种图像处理方法、装置、电子设备及存储介质。

背景技术

图像风格迁移(Photorealistic Style Transfer，PST)指自然图像之间的风格转换，指定一幅输入图像作为基础图像，也被称为内容图像，同时指定一幅图像作为希望得到的风格图像，算法在保证内容图像的内容同时，将内容图像的风格进行转换，最终输出的合成图像呈现出内容图像和风格图像的完美结合。在图像风格迁移的技术中，如何实现精准有效的迁移图像风格是整个技术的关键。

发明内容

鉴于上述问题，本申请提出了一种图像处理方法、装置、电子设备及存储介质。

第一方面，本申请实施例提供了一种图像处理方法，所述方法包括：获取第一图像、第二图像以及初始图像；利用预先训练的至少一个特征提取模型，提取所述第一图像的内容特征、所述第二图像的风格特征、所述初始图像的内容特征以及所述初始图像的风格特征；根据所述初始图像的内容特征与所述第一图像的内容特征的差异，确定内容损失函数的值，以及根据所述初始图像的风格特征中不同特征之间的相似度，与所述第二图像的风格特征中对应的不同特征之间的相似度的差异，确定风格损失函数的值；根据所述内容损失函数的值和所述风格损失函数的值确定总损失函数的值；根据所述总损失函数的值调整所述初始图像，直至得到目标图像，所述目标图像的内容特征与所述第一图像的内容特征匹配，所述目标图像的风格特征与所述第二图像的风格特征匹配。

第二方面，本申请实施例提供了一种图像处理装置，所述装置包括：图像获取模块、特征获取模块、第一损失获取模块、第二损失获取模块以及图像调整模块，其中，所述图像获取模块用于获取第一图像、第二图像以及初始图像；所述特征获取模块用于利用预先训练的至少一个特征提取模型，提取所述第一图像的内容特征、所述第二图像的风格特征、所述初始图像的内容特征以及所述初始图像的风格特征；所述第一损失获取模块用于根据所述初始图像的内容特征与所述第一图像的内容特征的差异，确定内容损失函数的值，以及根据所述初始图像的风格特征中不同特征之间的相似度，与所述第二图像的风格特征中对应的不同特征之间的相似度的差异，确定风格损失函数的值；所述第二损失获取模块用于根据所述内容损失函数的值和所述风格损失函数的值确定总损失函数的值；所述图像调整模块用于根据所述总损失函数的值调整所述初始图像，直至得到目标图像，所述目标图像的内容特征与所述第一图像的内容特征匹配，所述目标图像的风格特征与所述第二图像的风格特征匹配。

第三方面，本申请实施例提供了一种电子设备，包括：一个或多个处理器；存储器；一个或多个应用程序，其中所述一个或多个应用程序被存储在所述存储器中并被配置为由所述一个或多个处理器执行，所述一个或多个程序配置用于执行上述第一方面提供的图像处理方法。

第四方面，本申请实施例提供了一种计算机可读取存储介质，所述计算机可读取存储介质中存储有程序代码，所述程序代码可被处理器调用执行上述第一方面提供的图像处理方法。