[发明专利]一种图像风格化重建方法及装置

权利要求书

1.一种图像风格化重建方法，其特征在于，包括：

获取待转换的输入图像的第一边缘图像，以及获取预定的目标风格图像的第二边缘图像；

将第一边缘图像划分为r*r大小的第一图像块，以及将第二边缘图像划分为r*r大小的第二图像块，r为大于1的自然数；

获取第一边缘图像的每一第一图像块的相似图像块集合，该每一相似图像块集合中的元素为与该第一图像块相似的第二图像块；

根据所有的第一图像块，获取输入图像字典以及稀疏分解的稀疏系数；

根据所有第一图像块对应的相似图像块集合，获取目标图像字典；

根据所述输入图像字典、目标图像字典、稀疏分解的稀疏系数，获得重建出所述输入图像的目标风格的第三图像；

采用纹理迁移方式，生成所述输入图像的初始化风格图像；

将所述第三图像和所述初始化风格图像融合，得到用于输出的所述输入图像的重建风格化图像。

2.根据权利要求1所述的方法，其特征在于，所述获取待转换的输入图像的第一边缘图像，以及获取预定的目标风格图像的第二边缘图像，包括：

根据下述公式一，获取所述输入图像的滤波输出图像以及所述目标风格图像的滤波输出图像；

将所述输入图像减去所述输入图像的滤波输出图像，获得所述第一边缘图像，以及将所述目标风格图像减去所述目标风格图像的滤波输出图像，获得所述第二边缘图像；

其中，h_i＝Σ_jW_i,j(I)g_j 公式一

g为待滤波的图像，h是滤波后的输出图像，I是引导图像，输出图像中i位置的像素值h_i；i和j均为像素编号；W_i,j为核函数。

3.根据权利要求1所述的方法，其特征在于，获取第一边缘图像的每一第一图像块的相似图像块集合，包括：

根据下述公式二，确定所述第一边缘图像中每一第一图像块的相似图像块集合；

其中， 公式二

Diff(p,q)表示p图像块和q图像块之间的相似性，p为所述第一边缘图像中的第一图像块的任一个，q为所述第二边缘图像中的第二图像块的任一个；是梯度算子，η为系统参数。

4.根据权利要求1所述的方法，其特征在于，根据所有的第一图像块，获取输入图像字典，包括：

根据公式三，获取所述输入图像字典；

其中， 公式三

根据所有第一图像块对应的相似图像块集合，获取目标图像字典，包括：

根据公式四，获取所述目标图像字典；

其中， 公式四

D_p为输入图像字典，D_q为目标图像字典，P＝{p₁,p₂,…,p_n}为第一边缘图像中第一图像块的集合，Q＝{q₁,q₂,…,q_n}为第二边缘图像中与P对应的相似图像块的集合；Γ表示稀疏系数。

5.根据权利要求1所述的方法，其特征在于，根据所述输入图像字典、目标图像字典、稀疏分解的稀疏系数，获得重建出所述输入图像的目标风格的第三图像，包括：

根据公式五，获取所述第三图像

其中，z＝D_qγ 公式五

z为组成所述第三图像的图像块，D_q为目标图像字典，每一个图像块稀疏分解后的稀疏系数为γ。

6.根据权利要求1所述的方法，其特征在于，将所述第三图像和所述初始化风格图像融合，得到用于输出的所述输入图像的重建风格化图像，包括：

根据公式六，获取用于输出的所述输入图像的重建风格化图像；

其中， 公式六；

z为组成所述第三图像的图像块，z₀为初始化风格图像的图像块；α∈(0,1)为加权系数。