一种人脸遮挡图像数据集生成方法、装置、设备及介质

本发明公开了一种人脸遮挡图像数据集生成方法、装置、设备及介质，包括：获取包括多个人脸图像的原始数据集，采用边缘检测算法对原始数据集中的每个人脸图像提取边缘信息；根据边缘信息生成每个人脸图像对应的遮挡区域得到第一遮挡图像数据集；采用数据增强算法在原始数据集中的每个人脸图像上随机添加遮挡区域得到第二遮挡图像数据集；将第一遮挡图像数据集与第二遮挡图像数据集进行组合，得到用于训练人脸图像识别模型的目标遮挡图像数据集。本发明实施例的技术方案可以构建出包含人脸遮挡情况的训练数据集，提高人脸图像识别模型的泛化能力和鲁棒性。