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

说明书

本说明书公开了一种目标跟踪方法、装置、设备及存储介质，通过仅对待处理视频中的第一视频帧进行目标检测，得到目标对象，并根据目标对象在第一视频帧中的位置以及获取的待处理视频中各视频帧对应的运动矢量，确定目标对象在各第二视频帧中的位置，从而在待处理视频的各视频帧中对目标对象进行跟踪。可见，通过仅对待处理视频中的第一视频帧进行高精度的目标检测，在有限的算力条件下保证了目标检测的精度，同时，利用运动矢量确定目标对象在除第一视频帧之外的第二视频帧中的位置，实现对目标对象的跟踪，兼顾了目标跟踪的实时性。

技术领域

本说明书涉及计算机技术领域，尤其涉及一种目标跟踪方法、装置、设备及存储介质。

背景技术

随着视觉计算技术的发展，目标跟踪技术在视频转播、安防监控、无人驾驶等领域得到了广泛的应用。目标跟踪的主要任务是通过智能设备连续、实时、准确地定位视频序列中被跟踪的目标对象。其中，目标跟踪可采用密集的跟踪方法，即，使用目标检测模型，对视频中的每一帧图像均进行目标检测，确定被跟踪的目标对象在每一帧图像中的位置，从而实现对目标的跟踪。然而，由于智能设备(如智能摄像头)的算力有限，使用目标检测算法对视频图像进行运算的帧率远低于视频的帧率，因此无法通过对视频中的每一帧图像均进行目标检测运算，实现目标跟踪，导致目标跟踪的实时性降低，甚至无法对目标对象进行跟踪。

现有技术中，为了在有限的算力条件下在智能设备上实现目标跟踪，通常采用精简目标检测模型的方式，降低目标检测模型运算的时间，提高使用目标检测算法对视频图像进行运算的帧率，从而提高目标跟踪的实时性。

然而，对目标检测模型的精简会导致目标检测模型精度的下降，进而降低了目标跟踪的精度。

发明内容

本说明书提供一种目标跟踪方法、装置、设备及存储介质，以部分的解决现有技术存在的上述问题。

本说明书采用下述技术方案：

本说明书提供了一种目标跟踪方法，包括：

获得待处理视频，并确定所述待处理视频中的第一视频帧和第二视频帧；

通过预先训练的目标检测模型，对所述第一视频帧进行目标检测，以确定所述第一视频帧中的目标对象，以及所述目标对象在所述第一视频帧中的位置；并确定所述待处理视频中各视频帧对应的运动矢量；

根据所述目标对象在第一视频帧中的位置以及所述待处理视频中位于所述第一视频帧之后的第二视频帧对应的运动矢量，确定所述目标对象在所述第二视频帧中的位置；

根据所述目标对象在所述第一视频帧中的位置以及所述目标对象在所述第二视频帧中的位置，对所述目标对象进行跟踪。

可选地，确定所述待处理视频中的第一视频帧和第二视频帧，具体包括：

根据所述待处理视频的帧率以及预先训练的目标检测模型的帧率，确定抽帧的频率；

根据所述频率，对所述待处理视频进行抽帧，将抽取出的视频帧作为第一视频帧，将除第一视频帧之外的视频帧作为第二视频帧。

可选地，确定所述待处理视频中各视频帧对应的运动矢量，具体包括：

将所述待处理视频进行多尺度调整，得到多个不同尺度的待处理视频；

针对每个尺度的待处理视频，将该尺度的待处理视频中各视频帧进行分割，得到该尺度的各视频帧的像素块；并获取该尺度的各视频帧中每个像素块对应的运动矢量，作为该尺度的视频帧对应的运动矢量；一个视频帧中一个像素块对应的运动矢量用于表征该像素块从该视频帧到下一个视频帧的位移。

可选地，所述第一视频帧中的目标对象包括具有上下级关系的多个目标对象；其中，上一级目标对象在视频帧中覆盖的像素块包括下一级目标对象在视频帧中覆盖的像素块；