[发明专利]人体姿态识别中头部关键点的检测方法及装置

权利要求书

1.一种人体姿态识别中头部关键点的检测方法，其特征在于，包括：

提供待处理图像；

对所述待处理图像进行目标检测处理，以得到一个或多个第一头部检测框；

判断所述待处理图像中每个人体的头部姿态；

当所述人体的头部姿态为正面或背面时，分别将第一头部检测框各边的中点作为四个第一头部关键点；

当所述人体的头部姿态为左侧面时，将第一头部检测框右侧纵向边的中点、左下顶点以及上侧横向边的中点作为三个第一头部关键点；

当所述人体的头部姿态为右侧面时，将第一头部检测框左侧纵向边的中点、右下顶点以及上侧横向边的中点作为三个第一头部关键点。

2.如权利要求1所述的方法，其特征在于，还包括：

提供训练图像数据集，所述训练图像数据集包括多张训练图像；

分别采用人工方式在每张所述训练图像中标注第二头部关键点信息，并根据所述第二头部关键点信息计算确定第二头部检测框；

对每张所述训练图像进行目标检测处理，以得到一个或多个第三头部检测框；

根据所述第二头部检测框以及对应的所述第三头部检测框进行深度学习，以获得目标检测深度学习模型；

采用所述目标检测深度学习模型对所述待处理图像进行目标检测处理。

3.如权利要求1或2所述的方法，其特征在于，所述目标检测处理采用Faster-RCNN或SSD。

4.如权利要求2所述的方法，其特征在于，采用人工方式在每张所述训练图像中标注第二头部关键点信息包括：

当所述训练图像中人体为正面或背面时，将头顶、左耳、右耳和下巴作为四个第二头部关键点；当所述训练图像中人体为左侧面时，将头顶、左耳和下巴作为三个第二头部关键点；当所述训练图像中人体为右侧面时，将头顶、右耳和下巴作为三个第二头部关键点；

根据所述第二头部关键点信息计算确定第二头部检测框包括：

当所述训练图像中人体为正面或背面时，以四个第二头部关键点为第二头部检测框四个边的中点，从而确定所述第二头部检测框；

当所述训练图像中人体为左侧面时，以头顶对应的所述第二头部关键点作为第二头部检测框上侧横向边的中点、左耳对应的所述第二头部关键点作为第二头部检测框右侧纵向边的中点、下巴对应的所述第二头部关键点作为第二头部检测框左下顶点，从而确定所述第二头部检测框；

当所述训练图像中人体为右侧面时，以头顶对应的所述第二头部关键点作为第二头部检测框上侧横向边的中点、右耳对应的所述第二头部关键点作为第二头部检测框左侧纵向边的中点、下巴对应的所述第二头部关键点作为第二头部检测框右下顶点，从而确定所述第二头部检测框。

5.如权利要求1所述的方法，其特征在于，所述方法还包括：

当所述人体的头部姿态为左侧面或右侧面时，对第一头部检测框进行横向扩展处理，得到扩展后的第一头部检测框；

当所述人体的头部姿态为左侧面时，将第一头部检测框右侧纵向边的中点、左下顶点以及上侧横向边的中点作为三个头部关键点包括：当所述人体的头部姿态为左侧面时，将扩展后的第一头部检测框右侧纵向边的中点、左下顶点以及上侧横向边的中点作为三个第一初步头部关键点，对所述第一初步头部关键点进行与所述横向扩展处理相应的横向敛缩处理，以得到第一头部关键点信息；

当所述人体的头部姿态为右侧面时，将第一头部检测框左侧纵向边的中点、右下顶点以及上侧横向边的中点作为三个头部关键点包括：当所述人体的头部姿态为右侧面时，将扩展后的第一头部检测框左侧纵向边的中点、右下顶点以及上侧横向边的中点作为三个第一初步头部关键点，对所述第一初步头部关键点进行与所述横向扩展处理相应的横向敛缩处理，以得到第一头部关键点信息。

6.如权利要求1所述的方法，其特征在于，判断所述待处理图像中每个人体的头部姿态包括：

采用人体姿态热力图或回归关键点坐标计算所述待处理图像中的关键点信息；

根据计算的关键点信息获知人体的头部姿态。