[发明专利]行人图像识别方法、装置、电子设备及存储介质

权利要求书

1.一种行人图像识别方法，其特征在于，所述方法包括：

获取待检测行人图像，并对所述待检测行人图像进行预处理，该待检测行人图像包括：带遮罩的人脸图像；

将预处理后的图像输入至预先训练的人脸检测模型，以输出人脸图像，所述人脸检测模型基于历史行人图像和经过数据增强处理的历史行人图像进行训练，所述数据增加处理为对所述历史行人图像进行模板遮罩添加处理，该历史行人的人脸图像未带遮罩；

将输出的人脸图像输入至预先训练的多属性识别模型，以输出与该人脸图像相应的多个属性，所述多属性识别模型基于人脸图像的多粒度特征识别人脸图像的属性。

2.根据权利要求1所述的方法，其特征在于，通过如下方式训练所述人脸检测模型：

获取历史行人图像，并根据预定算法确定所述历史行人图像中的历史人脸图像中的关键点；

根据所述历史人脸图像中的关键点和预先构建的模板遮罩库对所述历史人脸图像进行数据增强处理，以生成历史模板遮罩人脸图像；

根据所述历史行人图像和所述历史模板遮罩人脸图像训练所述人脸检测模型。

3.根据权利要求2所述的方法，其特征在于，根据所述历史人脸图像中的关键点和预先构建的模板遮罩库对所述历史人脸图像进行数据增强处理，以生成历史模板遮罩人脸图像包括：

根据所述历史人脸图像中的关键点位置和关键点之间的距离确定该历史人脸的倾斜角度；

根据所述历史人脸图像中的关键点和该历史人脸的倾斜角度从所述模板遮罩库选择模板遮罩；

将选择的模板遮罩添加到所述历史人脸图像上，以生成所述历史模板遮罩人脸图像。

4.根据权利要求2所述的方法，其特征在于，通过如下方式构建所述模板遮罩库：

获取初始遮罩图像；

根据预先确定的人脸图像中的关键点位置对所述初始遮罩图像进行标定处理，以生成透明背景遮罩；

对所述透明背景遮罩进行几何变换和色彩调整处理，以生成不同类别、色彩和角度的模板遮罩；

根据生成的不同类别、色彩和角度的模板遮罩构建模板遮罩库。

5.根据权利要求2所述的方法，其特征在于，所述多属性识别模型包括多粒度网络，通过如下方式训练所述多属性识别模型：

获取已标注属性的历史行人图像和已标注人脸区域的历史行人图像；

基于图神经网络技术，并根据所述已标注属性的历史行人图像、已标注人脸区域的历史行人图像和所述历史模板遮罩人脸图像训练所述多属性识别模型。

6.根据权利要求5所述的方法，其特征在于，将输出的人脸图像输入至预先训练的多属性识别模型，以输出与该人脸图像相应的多个属性包括：

将输出的人脸图像输入至所述预先训练的多属性识别模型中的多粒度网络，以提取该人脸图像的多粒度特征；

基于图神经网络技术，对所述多粒度特征之间进行关联建模操作，以输出该人脸图像相应的多个属性。

7.一种行人图像识别装置，其特征在于，所述装置包括：

待检测图像获取单元，用于获取待检测行人图像，并对所述待检测行人图像进行预处理，该待检测行人图像包括：带遮罩的人脸图像；

人脸图像输出单元，用于将预处理后的图像输入至预先训练的人脸检测模型，以输出人脸图像，所述人脸检测模型基于历史行人图像和经过数据增强处理的历史行人图像进行训练，所述数据增加处理为对所述历史行人图像进行模板遮罩添加处理，该历史行人的人脸图像未带遮罩；

属性识别单元，用于将输出的人脸图像输入至预先训练的多属性识别模型，以输出与该人脸图像相应的多个属性，所述多属性识别模型基于人脸图像的多粒度特征识别人脸图像的属性。