[发明专利]目标跟踪方法、装置、电子设备及存储介质

说明书

本申请实施例公开了一种目标跟踪方法、装置、电子设备及存储介质，该方法包括：获取目标视频的当前图像帧中至少一个目标对象的第一目标框；对第一目标框进行重叠区域标记，得到第一目标框的遮挡情况；根据第一目标框的遮挡情况和目标对象的类别获取当前图像帧的光流跟踪点集；对光流跟踪点集进行稀疏光流计算，得到光流跟踪点集中每个光流跟踪点在当前图像帧的下一图像帧中对应的目标跟踪点；根据目标跟踪点获取目标对象在下一图像帧中的第二目标框。本申请实施例有利于提高对视频中的多目标进行跟踪的效率和效果。

技术领域

本申请涉及视频目标跟踪技术领域，尤其涉及一种目标跟踪方法、装置、电子设备及存储介质。

背景技术

随着机器视觉理论与技术的发展，对视频内容进行识别与理解成为研究的热点，尤其在基于视频的单个目标跟踪相关产品落地后，对视频中的多目标进行跟踪的需求也愈发强烈。当前基于视频的目标跟踪方法较多，例如：根据目标中心点距离、目标区域交并比等物理指标将相邻视频帧中的目标进行关联的方法，但这些方法在某些场景中并不适用；再者是现在应用较为普遍的光流跟踪法，但其多用于对视频图像中单个目标进行跟踪，且采用均匀选取跟踪点的方式，难免会使背景像素点也被当作跟踪点，可见，目前的光流跟踪法在进行多目标跟踪时，跟踪效率较差且跟踪效果并不显著。

发明内容

为解决上述问题，本申请提供了一种目标跟踪方法、装置、电子设备及存储介质，有利于提高对视频中的多目标进行跟踪的效率和效果。

本申请实施例第一方面提供了一种目标跟踪方法，该目标跟踪方法包括：

获取目标视频的当前图像帧中至少一个目标对象的第一目标框；

对所述第一目标框进行重叠区域标记，得到所述第一目标框的遮挡情况；

根据所述第一目标框的遮挡情况及所述目标对象的类别获取所述当前图像帧的光流跟踪点集；

对所述光流跟踪点集进行稀疏光流计算，得到所述光流跟踪点集中每个光流跟踪点在所述当前图像帧的下一图像帧中对应的目标跟踪点；

根据所述目标跟踪点获取所述目标对象在所述下一图像帧中的第二目标框。

结合第一方面，在一种可能的示例中，所述根据所述第一目标框的遮挡情况及所述目标对象的类别获取所述当前图像帧的光流跟踪点集，包括：

根据每个所述第一目标框的遮挡情况及每个所述第一目标框对应的所述目标对象的类别，从每个所述第一目标框的预设区域选取所述光流跟踪点；

将每个所述第一目标框选取出的所述光流跟踪点加入同一集合中，得到所述光流跟踪点集。

结合第一方面，在一种可能的示例中，所述根据每个所述第一目标框的遮挡情况及每个所述第一目标框对应的所述目标对象的类别，从每个所述第一目标框的预设区域选取所述光流跟踪点，包括：

对于属于第一类目标对象的所述第一目标框，若该第一目标框无遮挡，则以该第一目标框的中心为中心、以第一预设比例对该第一目标框进行缩小，得到第一选取窗口，在所述第一选取窗口内选取所述光流跟踪点；若所述第一选取窗口存在遮挡，则在所述第一选取窗口未被遮挡的区域选取所述光流跟踪点；

对于属于第二类目标对象的所述第一目标框，若该第一目标框无遮挡，则在该第一目标框的第一预设高度处和第二预设高度处选取所述光流跟踪点；若所述第一预设高度处或所述第二预设高度处存在遮挡，则在所述第一预设高度处和所述第二高度处未被遮挡的区域选取所述光流跟踪点；

对于属于第三类目标对象的所述第一目标框，若该第一目标框无遮挡，则以该第一目标框的中心为中心、以第二预设比例对该第一目标框进行缩小，得到第二选取窗口，在所述第二选取窗口内选取所述光流跟踪点；若所述第二选取窗口存在遮挡，则在所述第二选取窗口未被遮挡的区域选取所述光流跟踪点。