[发明专利]一种基于深度学习的人脸图像去遮挡方法

说明书

本发明公开了一种基于深度学习的人脸图像去遮挡方法，包括以下步骤：(1)自动对原始图片添加特定样式与分布的遮挡，模拟真实遮挡分布，制作训练数据集并对输入数据进行预处理；(2)设计基于全卷积网络的神经子网络N1学习遮挡的分布与深浅；(3)设计神经子网络N2对遮挡图像进行自动修复与平滑；(4)迭代训练子网络N1，N2，进行多任务学习，使得网络自动检测出遮挡并输出修复后的与原图大小相同的图像。本发明能够检测出特定形状与分布的遮挡，并且通过网络进行像素级别的自动修复与平滑，完成未遮挡部分的图像细节保持和遮挡部分的自动修复，使得后续的人脸识别正确率得到极大的提高，并且大大提高了网络的优化速度。

技术领域

本发明涉及一种图像去遮挡方法，尤其涉及一种基于深度学习的人脸图像去遮挡方法。

背景技术

随着计算机视觉技术的发展，人脸识别在家庭娱乐，安防系统，视频监控等领域得到广泛应用。人脸识别的各个步骤都需要输入人脸图像满足一定要求：人脸图像无明显噪声，遮挡，人脸图像清晰度，质量较好。目前，人脸图像去噪遮挡主要关注在去除干扰信息的同时，减少图像细节的丢失。

基于传统方法的图像去遮挡和修复主要分为两个方向：基于图像纹理分析技术和基于局部插值的图像修复方法。

基于传统纹理分析技术的图像去噪方法，其模型设计复杂，速度慢，效率低，容易带来图像细节模糊问题。而基于局部插值的图像修复方法，未使用到图像的全局信息，容易带来图像不平滑问题。对于缺少区域比较大的场景，效果较差，基于插值图像的修复方法需要事先知道噪声位置。

发明内容

为了解决上述问题中的不足之处，本发明提供了一种基于深度学习的人脸图像去遮挡方法。

为了解决以上技术问题，本发明采用的技术方案是：一种基于深度学习的人脸图像去遮挡方法，包括以下步骤：

(1)自动对原始图片添加特定样式与分布的遮挡，模拟真实遮挡分布，制作训练数据集并对输入数据进行预处理；

具体的预处理步骤包括但不限于将输入图像进行归一化、尺度缩放、数据增广；并且通过批处理操作将图片像素值进行归一化，便于后续计算与网络收敛；

(2)设计基于全卷积网络的神经子网络N1学习遮挡的分布与深浅；

其中，网络的前端部分较浅的神经网络的层通过对原始输入图片以级联卷积的方式计算得到的特征图，特征图分布反应了图像的基础信息，不同的卷积核提取出不同的特征图，并将边缘信息作为后续网络的输入；

(3)设计神经子网络N2对遮挡图像进行自动修复与平滑；

(3.1)深度神经网络N2采用与深度神经网络N1相同的前半部，实现参数共享，N2的核心思想是提取N1的末端的输入特征图，经过N1阶段的学习后，此时的特征图已比较接近遮挡分布，将该特征图和原始遮挡图片叠加，将叠加值作为初始修复状态；

(3.2)获取初始修复状态后，神经子网络N2的后半部采用不带下采样操作的卷积单元，包括但不局限于两层或者多层的卷积层，多个卷积核以滑动窗口的形式在步骤(3.1)得到的初始叠加值上计算新的特征图，输出以无遮挡的真值图像进行约束；

(3.3)然后采用欧式损失进行有监督约束，训练采用的算法是BackPropagation反向回传算法，通过回传的损失层误差来更新每一层的参数，使得网络参得到更新，最终得到收敛；

(4)迭代训练子网络N1，N2，进行多任务学习；

(4.1)首先完成遮挡的检测任务，任务的主要实施部分为网络结构N1，输入数据是有遮挡的原始图像，标签是已知的遮挡的分布图，目标是在N1结构的末端在测试阶段自动学习出遮挡的分布，使得神经网络N1收敛到良好状态。