[发明专利]一种人脸关键点检测模型的生成方法及相关设备

权利要求书

1.一种人脸关键点检测模型的生成方法，其特征在于，包括：

预设网络模型根据训练集中人脸图片，生成所述人脸图片对应的预测关键点位置信息组和预测关键点热度图，其中，所述训练集包括多组训练样本，每组训练样本包括人脸图片以及人脸图片对应的关键点热度图和关键点位置信息；

所述预设网络模型根据所述人脸图片对应的预测关键点位置信息组和所述预测关键点热度图，以及所述人脸图片对应的关键点热度图和关键点位置信息，对模型参数进行修正，并继续执行所述根据所述训练集中人脸图片，生成所述人脸图片对应的预测关键点位置信息组和预测关键点热度图的步骤，直至所述预设网络模型的训练情况满足预设条件，以得到所述人脸关键点检测模型。

2.根据权利要求1所述的人脸关键点检测模型的生成方法，其特征在于，所述预设网络模型根据所述人脸图片对应的预测关键点位置信息组和所述预测关键点热度图，以及所述人脸图片对应的关键点热度图和关键点位置信息，对模型参数进行修正的步骤包括：

根据所述人脸图片对应的预测关键点位置信息组和所述关键点位置信息，计算出位置损失值；

根据所述预测关键点热度图、关键点热度图计算出热度图损失值；

根据所述位置损失值和热度图损失值计算出总损失值；

根据所述总损失值对模型参数进行修正。

3.根据权利要求1所述的人脸关键点检测模型的生成方法，其特征在于，所述预设网络模型包括：特征图提取模块、融合图生成模块和预测模块；

所述预设网络模型根据训练集中人脸图片，生成所述人脸图片对应的预测关键点位置信息组和预测关键点热度图的步骤包括：

将所述人脸图片输入所述特征图提取模块，通过所述特征图提取模块得到所述人脸图片对应的下采样特征图组，所述下采样特征图组包括多张预设尺寸不同的下采样特征图；

将所述下采样特征图组输入所述融合图生成模块，通过所述融合图生成模块得到融合图组；其中，所述融合图组包括多张融合图，每张融合图对应一张下采样特征图，且各张融合图对应的下采样特征图不同；

将所述下采样特征图组中指定下采样特征图和所述融合图组输入到预测模块，通过所述预测模块得到所述人脸图片对应的预测关键点位置信息组和预测关键点热度图。

4.根据权利要求3所述的人脸关键点检测模型的生成方法，其特征在于，所述特征图提取模块包括：k层下采样单元，其中，k为正整数；

将所述人脸图片输入所述特征图提取模块，通过所述特征图提取模块得到所述人脸图片对应的下采样特征图组的步骤包括：

将所述人脸图片输入第一层下采样单元，通过所述第一层下采样单元输出第一下采样特征图；

将第二层下采样单元作为目标下采样单元，将所述第一下采样特征图作为目标特征图；

将所述目标特征图分别输入到所述目标下采样单元和所述融合图生成模块，通过所述目标下采样单元输出第二下采样特征图；

将所述目标下采样单元输出的第二下采样特征图作为目标特征图，并将所述目标下采样单元的后一层下采样单元作为目标下采样单元；

继续执行将目标特征图分别输入到所述目标下采样单元和所述融合图生成模块的步骤，直至通过第k层下采样单元输出第k下采样特征图，得到所述特征图提取模块输出的所述下采样特征图组。

5.根据权利要求3所述的人脸关键点检测模型的生成方法，其特征在于，所述融合图生成模块包括关键点信息生成单元和融合单元；

所述将所述下采样特征图组输入所述融合图生成模块，通过所述融合图生成模块得到融合图组的步骤包括：

将第k层下采样单元输出的所述第k下采样特征图输入到所述关键点信息生成单元，通过所述关键点信息生成单元得到与所述第k下采样特征图对应的关键点位置特征图；其中，所述关键点位置特征图包括预测出的关键点坐标信息；

将所述关键点位置特征图和所述下采样特征图组中除所述指定下采样特征图之外的其他下采样特征图输入到所述融合单元，通过所述融合单元输出所述融合图组。