一种基于KPNet的人体颈部关键点识别方法及系统

    公开(公告)号:CN115346241A

    公开(公告)日:2022-11-15

    申请号:CN202210970994.4

    申请日:2022-08-14

    摘要: 本发明公开了一种基于KPNet的人体颈部关键点识别方法及系统。方法包括:S1、采集包括人体颈部关键点的若干图像;S2、构建包括人体颈部关键点的训练数据集和验证数据集;S3、对CenterNet网络进行改进;S4、对改进后的CenterNet网络采用步骤S2中的训练数据集进行训练,得出KPNet网络;S5、通过相机拍摄得到的实际获取人体颈部图像输入到网络进行关键点检测;S6、将每个预测特征图的像素值以二维正态分布,取二维正态分布的峰值作为预测关键点,并将该预测关键点结合相机拍摄到的人体颈部点云信息,得出关键点的坐标信息。本发明通过改进网络热图处理输出需要的关键点,相应的其数据集标签也从锚框变成了关键点。