[发明专利]基于脸部的图像重建方法、装置、电子设备及存储介质

权利要求书

1.一种基于脸部的图像重建方法，其特征在于，所述方法包括：

获取脸部图像；

对所述脸部图像进行多次图像超分辨率重建，直至生成符合预设图像尺寸的重建后图像，该符合预设图像尺寸的重建后图像即为目标超分图像；

通过以下方式对所述脸部图像或进行图像超分辨率重建后不符合预设图像尺寸的重建后图像进行图像超分辨率重建：

对所述脸部图像或重建后图像进行特征提取，得到脸部特征图；

对所述脸部特征图进行上采样，得到脸部上采样图，所述脸部上采样图的尺寸大于所述脸部特征图的尺寸；

对所述脸部上采样图进行脸部边缘估计，得到脸部边缘图；

对所述脸部边缘图与所述脸部上采样图进行融合，得到重建后图像。

2.根据权利要求1所述的方法，其特征在于，在每一次图像超分辨率重建的过程中，所述对所述脸部上采样图进行脸部边缘估计，得到脸部边缘图，包括：

基于预设的高斯核，对所述脸部上采样图进行脸部边缘估计，得到脸部边缘图；

在每一次图像超分辨率重建过程中使用的高斯核的尺寸大于上一次图像超分辨率重建过程中使用的高斯核的尺寸。

3.根据权利要求1所述的方法，其特征在于，所述对所述脸部上采样图进行脸部边缘估计，得到脸部边缘图，包括：

对所述脸部上采样图进行特征提取，得到特征提取图；

对所述特征提取图进行归一化，得到归一化图像；

对所述归一化图像进行去线性化，得到去线性化图像；

计算所述去线性化图像中每个通道的重要性得分，基于每个通道的重要性得分，对所述去线性化图像中每个通道的特征进行加权，得到通道增强后图像；

对所述通道增强后图像进行边缘估计，得到脸部边缘图。

4.根据权利要求3所述的方法，其特征在于，所述对所述通道增强后图像进行边缘估计，得到脸部边缘图，包括：

对所述通道增强后图像中预设脸部区域内包含的各脸部子区域分别进行关键点检测，得到包含各脸部子区域内关键点的脸部关键点图；

对所述脸部关键点图进行边缘估计，得到脸部边缘图。

5.根据权利要求4所述的方法，其特征在于，所述对所述脸部关键点图进行边缘估计，得到脸部边缘图，包括：

对所述脸部关键图进行预设数量次的下采样处理，得到下采样图像；其中，每次下采样处理的结果作为下一次进行下采样处理所处理的图像；

对每一次下采样处理前的图像进行特征提取，得到与该图像尺寸相同的特征图；

对所述下采样图像进行所述预设数量次的上采样处理，并且对每一次上采样处理的结果以及与该结果尺寸相同的特征图进行融合，得到融合图像，该融合图像作为下一次进行上采样处理所处理的图像；其中，将最终得到的融合图像作为脸部边缘图。

6.根据权利要求1-5中任一项所述的方法，其特征在于，在所述对所述脸部图像进行特征提取，得到脸部特征图的步骤之后，还包括：

计算所述脸部特征图中每个像素点的重要性得分，基于每个像素点的重要性得分，对所述脸部特征图中每一像素点的特征进行加权，得到空间增强后图像；

所述对所述脸部特征图进行上采样，得到脸部上采样图，包括：

对所述空间增强后图像进行上采样，得到脸部上采样图。

7.根据权利要求1-5中任一项所述的方法，其特征在于，所述方法还包括：

在图像超分辨率重建的过程中，针对每一次的不符合预设图像尺寸的重建后图像，对该次重建后图像与该次得到的脸部上采样图进行融合，将融合后的结果作为下一次进行图像超分辨率重建所处理的图像。