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

摘要

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

主权项

1.一种基于深度学习的人脸图像去遮挡方法，其特征在于，包括以下步骤：(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收敛到良好状态；(4.2)然后完成图像修复任务，任务的主要实施部分为网络结构N2，输入数据是有遮挡的原始图像，标签是未遮挡的真值图像，目标是在N2结构的末端在测试阶段自动学习出修复遮挡之后的图像，并通过图像级别像素值的回归，进一步减小学习率，促进网络收敛的步骤，使得网络自动检测出遮挡并输出修复后的与原图大小相同的图像。