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

说明书

本申请公开了一种目标跟踪方法、装置、存储介质及电子设备，属于人工智能技术领域。所述方法包括：获取摄像头采集的视频流，对所述视频流进行解码得到多帧图像；对于待处理的当前帧图像，当所述当前帧图像为第一类型图像时，对所述当前帧图像进行目标检测，得到被检测目标在所述当前帧图像中的位置信息；当所述当前帧图像为第二类型图像时，基于上一帧图像的运动向量，预测所述上一帧图像中的被检测目标在所述当前帧图像中的位置信息。本申请提供的目标跟踪方法既利用到了解码出来的全部图像的信息，又避免了逐帧图像进行目标检测带来的速率较慢的问题，不但确保了实时的跟踪速度，而且还确保了得到的跟踪结果的精确度。

技术领域

本申请涉及人工智能技术领域，特别涉及一种目标跟踪方法、装置、存储介质及电子设备。

背景技术

计算机视觉技术作为人工智能的核心，其应用范围目前已经遍及多个领域。比如，目标跟踪作为计算机视觉技术的一个重要研究方向，在当下有着十分广泛的应用，可应用在诸如人机交互、无人驾驶等场景下。

以单摄像头多目标跟踪为例，相关技术在进行目标跟踪时，首先将摄像头采集到的视频流解码为多帧图像；之后，基于目标检测算法对解码出来的每一帧图像均进行目标检测，进而根据对每一帧图像的目标检测结果实现目标跟踪。

由于需要对每一帧图像均进行目标检测，因此较为耗时，尤其是在面对高分辨率图像时，用于目标检测的图像越大，检测速度便越慢，这会严重影响目标跟踪的性能，很难做到实时跟踪。而为了达到实时跟踪的效果，相关技术采用的在解码出来的全部帧图像中抽取少数进行目标检测而剩余图像不做任何处理的方式，由于未利用到全部帧图像的信息，又会严重制约跟踪结果的准确性。基于此，如何进行目标跟踪，以在提升跟踪速度的同时又能够确保跟踪精确度，成为了本领域技术人员亟待解决的一个问题。

发明内容

本申请实施例提供了一种目标跟踪方法、装置、存储介质及电子设备，能够在提升跟踪速度的同时又能够确保跟踪精确度。所述技术方案如下：

一方面，提供了一种目标跟踪方法，所述方法包括：

获取摄像头采集的视频流，对所述视频流进行解码得到多帧图像；

对于待处理的当前帧图像，当所述当前帧图像为第一类型图像时，对所述当前帧图像进行目标检测，得到被检测目标在所述当前帧图像中的位置信息；

当所述当前帧图像为第二类型图像时，基于上一帧图像的运动向量，预测所述上一帧图像中的被检测目标在所述当前帧图像中的位置信息。

另一方面，提供了一种目标跟踪装置，所述装置包括：

获取模块，用于获取摄像头采集的视频流；

解码模块，用于对所述视频流进行解码得到多帧图像；

处理模块，用于对于待处理的当前帧图像，当所述当前帧图像为第一类型图像时，对所述当前帧图像进行目标检测，得到被检测目标在所述当前帧图像中的位置信息；当所述当前帧图像为第二类型图像时，基于上一帧图像的运动向量，预测所述上一帧图像中的被检测目标在所述当前帧图像中的位置信息。

在一种可能的实现方式中，当所述当前帧图像为第一类图像时，所述处理模块，还用于在对所述当前帧图像进行目标检测之后，对所述当前帧图像中的被检测目标进行关键点检测；基于得到的关键点检测结果和所述当前帧图像的运动向量，预测所述当前帧图像中的被检测目标在下一帧图像中的位置信息。

在一种可能的实现方式中，所述当前帧图像中的被检测目标为一个或多个，所述处理模块，还用于对于每个被检测目标，基于所述被检测目标的关键点检测结果，将所述被检测目标划分为多个区域；基于所述当前帧图像的运动向量，获取所述被检测目标的每个区域的运动向量偏置；基于所述每个区域的运动向量偏置，预测所述被检测目标在所述下一帧图像中的位置信息。