[发明专利]一种物品追踪方法、装置、设备及存储介质

权利要求书

1.一种物品追踪方法，其特征在于，所述方法包括：

获取从待检测视频中提取的多个视频帧；

针对每个所述视频帧，分别进行物品位置检测和物品属性识别，得到所述视频帧中各物品的物品位置信息和物品属性信息；所述物品属性信息包括物品品种和/或物品型号；

基于所述物品位置信息和物品属性信息，确定所述多个视频帧中对应同一目标物品的待追踪物品；

基于所述目标物品分别在所述多个视频帧中对应的待追踪物品的物品位置信息，生成所述目标物品的移动轨迹。

2.根据权利要求1所述的方法，其特征在于，所述基于所述物品位置信息和物品属性信息，确定所述多个视频帧中对应同一目标物品的待追踪物品，包括：

针对所述多个视频帧中的相邻视频帧，确定所述相邻视频帧中第一视频帧中的至少一个第一物品和所述相邻视频帧中第二视频帧中的至少一个第二物品；

针对每个所述第一物品，基于所述第一物品在所述第一视频帧中的物品位置信息和物品属性信息，以及每个所述第二物品在所述第二视频帧中的物品位置信息和物品属性信息，确定所述第一物品和每个所述第二物品之间的相似度；

根据所述相似度，确定对应同一目标物品的所述第一物品和所述第二物品，所述待追踪物品包括所述第一物品和所述第二物品。

3.根据权利要求2所述的方法，其特征在于，所述基于所述第一物品在所述第一视频帧中的物品位置信息和物品属性信息，以及每个所述第二物品在所述第二视频帧中的物品位置信息和物品属性信息，确定所述第一物品和每个所述第二物品之间的相似度，包括：

基于所述第一物品在所述第一视频帧中的物品属性信息和所述第二物品在所述第二视频帧中的物品属性信息，计算所述第一物品与所述第二物品的第一相似度；

基于所述第一物品在所述第一视频帧中的物品位置信息和所述第二物品在所述第二视频帧中的物品位置信息，计算所述第一物品与所述第二物品的第二相似度；

按照所述第一相似度对应的第一权重和所述第二相似度对应的第二权重，计算所述第一物品与所述第二物品的相似度。

4.根据权利要求3所述的方法，其特征在于，所述物品位置信息包括物品检测框信息，所述物品检测框信息包括物品检测框的高度和宽度，以及中心位置；所述第二相似度包括尺寸相似度和中心位置相似度；

所述基于所述第一物品在所述第一视频帧中的物品位置信息和所述第二物品在所述第二视频帧中的物品位置信息，计算所述第一物品与所述第二物品的第二相似度，包括：

基于所述第一物品的物品检测框的高度和宽度，以及所述第二物品的物品检测框的高度和宽度，计算所述第一物品和所述第二物品的尺寸相似度；

基于所述第一物品的检测框中心位置和所述第二物品的检测框中心位置，计算所述第一物品和所述第二物品的中心位置相似度。

5.根据权利要求4所述的方法，其特征在于，所述按照所述第一相似度对应的第一权重和所述第二相似度对应的第二权重，计算所述第一物品与所述第二物品的相似度，包括：

按照所述第一相似度对应的第一权重、所述尺寸相似度对应的第三权重和所述中心位置相似度对应的第四权重，计算所述第一物品与所述第二物品的相似度。

6.根据权利要求2述的方法，其特征在于，所述根据所述相似度，确定对应同一目标物品的所述第一物品和所述第二物品，包括：

在每个所述视频帧中分别检测出多个物品的情况下，对多个所述第一物品和多个所述至少一个第二物品进行配对组合，生成多个物品追踪链；所述物品追踪链中包含多个物品映射对；每个所述物品映射对中包含一个所述第一物品和一个所述第二物品；

针对每个所述物品追踪链，基于所述物品追踪链中多个物品映射对所对应的所述相似度，确定所述物品追踪链对应的相似度和值；

基于所述多个物品追踪链分别对应的相似度和值，从所述多个物品追踪链中选择目标物品追踪链，确定该目标物品追踪链中的各物品映射对所包含的所述第一物品和所述第二物品为同一目标物品。