一种基于条件生成对抗网络的人脸去遮挡方法

摘要

本发明公开了一种基于条件生成对抗网络的人脸去遮挡方法，包括如下步骤：准备人脸图像数据集和可能出现在面部的遮挡图像集，对每一张人脸图像随机加遮挡后，得到原始人脸图像、有遮挡人脸图像、遮挡图像组成的三元组数据集；将有遮挡人脸图像作为输入，遮挡图像作为目标图像，通过训练第一个条件生成对抗网络实现遮挡检测的功能；将有遮挡人脸图像和第一个条件生成对抗网络检测出的遮挡图像作为输入，原始人脸图像作为目标图像，通过训练第二个条件生成对抗网络实现人脸遮挡移除的功能。本发明在面部遮挡检测的基础上进行面部遮挡移除，不仅增强了对任意遮挡的识别能力，同时也借助学得的遮挡信息来更好地移除遮挡，使模型更具有鲁棒性。