[发明专利]一种基于行人检测的关节点跟踪方法及系统

权利要求书

1.一种基于行人检测的关节点跟踪方法，其特征在于，包括如下步骤：

S1、获取目标区域的RGB图像，并对所述RGB图像进行检测以确定所述目标区域中行人的第一2D边界框以及关节点坐标；

S2、根据步骤S1中确定的所述关节点坐标，利用所述关节点的连接关系以确定所述关节点是否属于同一个人，并根据所述关节点坐标计算所述关节点的第一2D矩形框；

S3、根据步骤S1中所获得的第一2D边界框对行人进行跟踪，得到行人的id；

S4、根据步骤S3中所获得的行人的id确定关节点的id。

2.如权利要求1所述的基于行人检测的关节点跟踪方法，其特征在于：在步骤S1中,利用特征提取方法提取所述RGB图像中的全局特征，根据所述全局特征输出行人检测信息和关节点的坐标信息。

3.如权利要求1所述的基于行人检测的关节点跟踪方法，其特征在于：步骤S3中包括:

利用所述第一2D边界框构建运动模型，估算下一帧图像中目标行人的第二2D边界框；

将估算的所述第二2D边界框与检测到的下一帧图像中目标行人的2D边界框进行匹配；

根据匹配结果确定行人的id。

4.如权利要求1所述的基于行人检测的关节点跟踪方法，其特征在于：步骤S4中，将所述第一2D边界框与所述第一2D矩形框进行匹配，并根据匹配结果利用所述行人的id确定所述关节点的id。

5.如权利要求4所述的基于行人检测的关节点跟踪方法，其特征在于：通过匹配计算同一帧时刻行人的所述第一2D边界框与关节点的所述第一2D矩形框的交并比，根据匹配结果将行人的所述第一2D边界框与关节点的所述第一2D矩形框进行关联，将每一个行人的id对应每一组关节点的id，从而完成关节点的跟踪。

6.如权利要求1所述的基于行人检测的关节点跟踪方法，其特征在于：采用共同训练的方法获取行人检测学习参数和关节点检测学习参数，以利用所述RGB图像确定行人的所述第一2D边界框和所述关节点坐标；或者，

采用分步训练方法获取行人检测学习参数和关节点检测学习参数，以利用所述RGB图像确定行人的所述第一2D边界框和所述关节点坐标。

7.一种基于行人检测的关节点跟踪系统，其特征在于，包括：相机拍摄模块、目标提取模块、边框确定模块、目标跟踪与匹配模块、以及关联模块；其中，

所述相机拍摄模块用于获取目标区域的RGB图像；

所述目标提取模块用于提取所述RGB图像中的全局特征，根据所述全局特征输出行人检测信息和关节点的坐标信息；

所述边框确定模块用于根据所述行人检测信息和所述关节点的坐标信息确定行人的第一2D边界框和关节点的第一2D矩形框；

所述目标跟踪与匹配模块基于所述第一2D边界框对目标行人进行跟踪，得到行人的id；

所述关联模块用于根据所述行人的id确定所述关节点的id。

8.如权利要求7所述的基于行人检测的关节点跟踪系统，其特征在于：所述目标跟踪与匹配模块利用行人的所述第一2D边界框与关节点的所述第一2D矩形框进行匹配计算，并将匹配结果传输至所述关联模块。

9.如权利要求8所述的基于行人检测的关节点跟踪系统，其特征在于：所述关联模块根据所述匹配结果将行人的所述第一2D边界框与关节点的所述第一2D矩形框进行关联，根据行人的id确定关节点的id。

10.一种计算机设备，所述计算机设备包括存储器、处理器以及存储在所述存储器上并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时至少实现权利要求1-6任一项所述的方法。