[发明专利]物体跟踪计数方法、装置、计算机设备和存储介质

说明书

本申请涉及一种物体跟踪计数方法、装置、计算机设备和存储介质，其中，方法包括：获取物体视频流，先对物体视频流进行物体检测，再进行基于常规跟踪算法的跟踪检测，以跟踪物体，针对检测得到的物体边界框数据和物体跟踪边界框数据采用卡尔曼滤波进行跟踪，最后基于卡尔曼跟踪得到的数据进行线性跟踪，计数运动至目的位置的物体数量。整个过程中，采用物体检测识别物体，通过常规跟踪算法确定在接下来物体的位置，采用卡尔曼滤波的方式获取物体的运动速度，其能够准确对物体进行线性跟踪，整个过程无需依赖复杂已训练的深度学习模型以及持续的基于特征的跟踪，实现过程简单，可以实现高效且准确的物体跟踪计数。

技术领域

本申请涉及图像处理技术领域，特别是涉及一种物体跟踪计数方法、装置、计算机设备和存储介质。

背景技术

随着图像处理技术的发展，目前图像处理技术已经广泛应用于生产和生活中给人们带来便利。例如图像处理技术可以应用于物体跟踪计数，采用图像采集设备采集物体图像，采用常规跟踪计数方式即可计数出在一定时间内物体的数量。

上述的物体跟踪计数方式最常见是应用于快递行业，在快递行业的中转场中，皮带机包裹流量估计具有巨大的应用意义，例如，可以用来计算中转场工作效率，可以用来提高设备和人员的使用能效等等。皮带机包裹流量估计最重要的工作是统计一段时间内包裹的数量，基于机器视觉的方法因安装方便，精度高，成本低而得到了广泛的应用。

然而，传统的物体跟踪计数方案无法兼顾计数精度和计数效率，实现过程复杂。

发明内容

基于此，有必要针对上述技术问题，提供一种准确且高效的物体跟踪计数方法、装置、计算机设备和存储介质。

一种物体跟踪计数方法，方法包括：

获取物体视频流；

对物体视频流进行物体检测，获取检测输出的物体边界框数据；

根据检测输出的物体边界框数据、采用预设跟踪算法跟踪物体，获取物体跟踪边界框数据；

根据检测输出的物体边界框数据以及物体跟踪边界框数据、采用卡尔曼滤波，得到卡尔曼跟踪边界框数据；

根据卡尔曼跟踪边界框数据进行线性跟踪，计数运动至目的位置的物体数量。

在其中一个实施例中，对物体视频流进行物体检测，获取检测输出的物体边界框数据包括：

对物体视频流进行物体检测，获取检测输出的物体初始边界框数据；

获取历史边界框数据，历史边界框数据为历史记录物体在物体检测时的边界框数据；

构建物体初始边界框数据与历史边界框数据之间的iou矩阵；

对iou矩阵做二分图匹配；

根据二分图匹配结果与预设约束阈值，更新历史边界框数据，得到检测输出的物体边界框数据。

在其中一个实施例中，根据二分图匹配结果与预设约束阈值，更新历史边界框数据，得到检测输出的物体边界框数据包括：

获取二分图匹配结果对应的匹配结果值；

若匹配结果值小于预设约束阈值，则判定检测到新物体，则在历史边界框数据中增加新物体对应的边界框数据，得到检测输出的物体边界框数据；

若匹配结果值不小于预设约束阈值，则判定检测到相同物体，更新历史边界框数据中相同物体对应的边界框数据，得到检测输出的物体边界框数据。

在其中一个实施例中，对iou矩阵做二分图匹配包括：

采用匈牙利算法对iou矩阵做二分图匹配。

在其中一个实施例中，对物体视频流进行物体检测包括：