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

权利要求书

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

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

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

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

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

2.根据权利要求1所述的方法，其特征在于，所述图像处理网络为可逆模型；

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

基于所述图像处理网络中各层级的前向计算过程，提取所述基础图像的图像内容特征以及所述参考图像的图像风格特征；

所述基于所述图像处理网络，将所述图像融合特征重建为目标图像，包括：

基于所述图像处理网络中各层级的逆向计算过程，将所述图像融合特征重建为所述目标图像；

其中，所述前向计算过程与所述逆向计算过程互为可逆计算。

3.根据权利要求2所述的方法，其特征在于，所述图像处理网络包括图像压缩层和多个流单元，每个所述流单元包括标准化层、可逆卷积层和耦合层；

所述基于所述图像处理网络中各层级的前向计算过程，对所述基础图像进行图像内容特征提取，包括：

对于未与所述图像压缩层连接的任意一个流单元，基于所述流单元的标准化层，对上一个流单元输出的内容特征张量进行线性变换，得到第一输出张量；

基于所述流单元的可逆卷积层，对所述第一输出张量进行卷积操作，得到第二输出张量；

基于所述流单元的耦合层，对所述第二输出张量进行切分，得到第一子张量和第二子张量；对所述第一子张量进行非线性变换，并将得到的非线性变换结果与所述第二子张量相加，得到第三子张量；对所述第一子张量和所述第三子张量进行特征拼接，得到第三输出张量；将所述第三输出张量输入下一个流单元。

4.根据权利要求3所述的方法，其特征在于，所述基于所述流单元的标准化层，对上一个流单元输出的内容特征张量进行线性变换，得到第一输出张量，包括：

将上一个流单元输出的内容特征张量中的每个元素分别与第一参数进行点乘，得到第一中间结果；

将所述第一中间结果与第二参数相加，得到所述第一输出张量。

5.根据权利要求3所述的方法，其特征在于，所述基于所述流单元的可逆卷积层，对所述第一输出张量进行卷积操作，得到第二输出张量，包括：

将所述第一输出张量中的每个元素分别与第一权重矩阵相乘，得到所述第二输出张量；

其中，所述第一输出张量和所述第二输出张量的通道数相同；所述第一权重矩阵的大小为c*c，c为正整数，c指代所述第一输出张量和所述第二输出张量的通道数。

6.根据权利要求2所述的方法，其特征在于，所述图像处理网络包括图像压缩层和多个流单元，每个所述流单元包括标准化层、可逆卷积层和耦合层；

所述基于所述图像处理网络中各层级的逆向计算过程，将所述图像融合特征重建为目标图像，包括：

对于未与所述图像压缩层连接的任意一个流单元，基于所述流单元的耦合层，对上一个流单元输出的融合特征张量进行切分，得到第四子张量和第五子张量；对所述第四子张量进行非线性变换，并将所述第五子张量与得到的非线性变换结果相减，得到第六子张量；对所述第四子张量和所述第六子张量进行特征拼接，得到第四输出张量；

基于所述流单元的可逆卷积层，对所述第四输出张量进行卷积操作的逆变换，得到第五输出张量；

基于所述流单元的标准化层，对所述第五输出张量进行线性变换的逆变换，得到第六输出张量；将所述第六输出张量输入下一个流单元。