[发明专利]一种多类别多目标的跟踪方法及系统

权利要求书

1.一种多类别多目标的跟踪方法，其特征在于，所述方法包括：

获取并解码视频流，得到视频序列RGB图像；

将所述视频序列RGB图像先进行图像特征预提取，所述图像特征预提取之后进行目标的候选框信息提取与Reid特征信息提取；

对所述目标的候选框信息与Reid特征信息关联匹配后输出。

2.根据权利要求1所述的跟踪方法，其特征在于，所述对所述目标的候选框信息提取与Reid特征信息关联匹配后输出包括对所述目标的候选框信息提取与Reid特征信息进行置信度得分检测，得到多类别目标的识别结果。

3.根据权利要求2所述的跟踪方法，其特征在于，所述对所述目标的候选框信息提取与Reid特征信息关联匹配后输出包括对Reid特征信息进行特征向量匹配检测，得到前后视频帧中为同一目标的识别结果。

4.根据权利要求3所述的跟踪方法，其特征在于，在得到所述多类别目标的识别结果与前后视频帧中为同一目标的识别结果之后进行目标信息输出，所述目标信息包括置信度得分、目标候选框的坐标信息以及每个目标的Reid特征向量。

5.根据权利要求4所述的跟踪方法，其特征在于，在所述进行目标信息输出之后进行信息保存，连续匹配跟踪目标，利用保存的信息使用卡尔曼滤波器对上一帧的目标进行预测，使用预测值与当前帧的目标信息作比较：

若当前帧的目标信息相似度与前一帧的预测值使用匈牙利算法能匹配上，则继续进行目标信息保存输出与目标状态更新；

若当前帧的目标信息相似度与前一帧的预测值使用匈牙利算法未能匹配上，则使用IOU计算两个目标框之间的交叠比，再进行一轮匈牙利算法匹配，若匹配，则将跟踪中目标状态进行更新。

6.根据权利要求5所述的跟踪方法，其特征在于，还包括若所述当前帧的目标信息相似度与前一帧的预测值使用匈牙利算法未能匹配上且进行两轮未匹配上，则对未匹配上的目标信息做状态更新。

7.一种多类别多目标的跟踪的系统，其特征在于，包括获取解码模块，图像特征提取模块，关联匹配输出模块，

所述获取解码模块用于获取并解码视频流，得到视频序列RGB图像；

所述图像特征提取模块用于将所述视频序列RGB图像先进行图像特征预提取，所述图像特征预提取之后同时进行目标的候选框信息提取与Reid特征信息提取；

所述关联匹配输出模块用于对所述目标的候选框信息提取与Reid特征信息关联匹配输出。

8.根据权利要求7的多类别多目标的跟踪的系统，其特征在于，还包括置信度得分检测单元，同一目标的识别单元，目标信息输出单元，连续匹配跟踪比较单元，

所述置信度得分检测单元用于对所述目标的候选框信息提取与Reid特征信息进行置信度得分检测得到多类别目标的识别结果；

所述同一目标的识别单元用于对Reid特征信息进行特征向量匹配检测得到前后视频帧中为同一目标的识别结果；

所述目标信息输出单元用于在所述得到所述多类别目标的识别结果与前后视频帧中为同一目标的识别结果之后进行目标信息输出，所述目标信息包括置信度得分，目标候选框的坐标信息，以及每个目标的Reid特征向量；

所述连续匹配跟踪比较单元用于在所述进行目标信息输出之后进行信息保存，连续匹配跟踪目标，利用保存的信息使用卡尔曼滤波器对上一帧的目标进行预测，使用预测值与当前帧的目标信息作比较。

9.一种多类别多目标的跟踪的设备，包括存储器、处理器、以及存储在所述存储器中并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时，实现如权利要求1至6任一项所述的方法中的各个步骤。