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

说明书

本申请公开了一种图像处理方法、装置、设备及存储介质，涉及计算机视觉技术。本申请可应用于云技术、人工智能、智慧交通等各种场景下。该图像处理方法包括：将基础图像和参考图像输入图像处理网络；基于图像处理网络，提取基础图像的图像内容特征以及参考图像的图像风格特征；基于参考图像的图像风格特征，在特征层面对基础图像的图像内容特征进行图像风格迁移，得到图像融合特征；基于图像处理网络，将图像融合特征重建为目标图像，目标图像具有基础图像的图像内容和参考图像的图像风格。本申请在图像处理过程中不会损失原始图像内容，进而可以生成图像内容丰富且具有参考图像风格的目标图像，提升了目标图像质量，确保了图像处理效果。

技术领域

本申请涉及计算机视觉技术，特别涉及一种图像处理方法、装置、设备及存储介质。

背景技术

计算机视觉技术涉及图像处理、图像识别、图像语义理解、图像检索、视频处理、视频语义理解等多个分支。其中，图像处理是指将一张图像转换成具有所希望特性的另一张图像。示例性地，图像风格迁移即是一种图像处理技术。

其中，图像风格迁移也称图像样式迁移，目的是将一张图像的风格(也称样式)迁移到另一张图像中。换言之，给定一张基础图像和一张参考图像，通过图像风格迁移，可以得到既保留原始图像内容又具有新样式的目标图像。示例性地，图像的风格包括但不限于色彩、纹理等。

然而，在图像风格迁移过程中，一旦损失基础图像的图像内容，在目标图像中便会出现图像内容部分缺失的现象，进而降低图像质量，影响图像处理效果。因此，亟需一种新的图像处理方法，以实现高质量的图像风格迁移。

发明内容

本申请实施例提供了一种图像处理方法、装置、设备及存储介质，该技术方案在整个图像处理过程中不会损失原始图像内容，能够生成图像内容丰富且具有参考图像风格的目标图像，提升了目标图像质量，进而确保了图像处理效果。该技术方案如下：

一方面，提供了图像处理方法，所述方法包括：

将基础图像和参考图像输入图像处理网络；

基于所述图像处理网络，提取所述基础图像的图像内容特征以及所述参考图像的图像风格特征；

基于所述参考图像的图像风格特征，在特征层面对所述基础图像的图像内容特征进行图像风格迁移，得到图像融合特征；

基于所述图像处理网络，将所述图像融合特征重建为目标图像，所述目标图像具有所述基础图像的图像内容和所述参考图像的图像风格。

在一些实施例中，所述图像处理网络的训练过程包括：

将样本基础图像和样本参考图像输入初始网络；基于所述初始网络，获取样本目标图像，所述样本目标图像具有所述样本基础图像的图像内容以及所述样本参考图像的图像风格；

基于所述样本基础图像和所述样本目标图像，构建图像内容损失函数；

基于所述样本参考图像和所述样本目标图像，构建图像风格损失函数；

根据所述图像内容损失函数和所述图像风格损失函数，对所述初始网络的网络参数进行更新，得到所述图像处理网络。

另一方面，提供了一种图像处理装置，所述装置包括：

特征提取模块，被配置为将基础图像和参考图像输入图像处理网络；基于所述图像处理网络，提取所述基础图像的图像内容特征以及所述参考图像的图像风格特征；

风格迁移模块，被配置为基于所述参考图像的图像风格特征，在特征层面对所述基础图像的图像内容特征进行图像风格迁移，得到图像融合特征；

图像重建模块，被配置为基于所述图像处理网络，将所述图像融合特征重建为目标图像，所述目标图像具有所述基础图像的图像内容和所述参考图像的图像风格。