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

权利要求书

1.一种视频目标追踪方法，其特征在于，所述方法包括：

获取待追踪视频；

将所述待追踪视频输入目标追踪模型，以得到所述待追踪视频对应的目标追踪结果，所述目标追踪模型用于执行如下处理：

针对所述待追踪视频的每一帧图像，确定所述图像对应的目标检测图像中待追踪目标对应的特征向量，所述目标检测图像包括所述待追踪目标；

将所述图像对应的特征图中的每一特征向量与所述目标检测图像中所述待追踪目标对应的所述特征向量进行第一相似度计算，并根据第一相似度计算结果，在所述特征图的所有特征向量中确定目标特征向量；

根据所述目标特征向量，确定所述图像中的待追踪目标。

2.根据权利要求1所述的方法，其特征在于，所述目标检测图像为包括所述待追踪目标的、所述图像的上一帧图像；或者

所述目标检测图像为包括所述待追踪目标的预设输入图像。

3.根据权利要求1所述的方法，其特征在于，所述根据第一相似度计算结果，在所述特征图的所有特征向量中确定目标特征向量，包括：

当所述目标检测图像包括N个所述待追踪目标时，在所述特征图的所有特征向量中，选取第一相似度计算结果最大的N个特征向量作为目标特征向量，所述N为正整数。

4.根据权利要求1所述的方法，其特征在于，所述目标追踪模型还用于根据预训练的位置向量参数确定所述图像中所有目标对应的特征向量，所述根据所述第一相似度计算结果，在所述特征图的所有特征向量中确定目标特征向量，包括：

当所述目标检测图像中包括N个所述待追踪目标时，在所述特征图的所有特征向量中，选取第一相似度计算结果最大的N个特征向量作为相似特征向量，所述N为正整数；

对所述图像中所有目标对应的特征向量和所述N个相似特征向量进行去重处理，以得到目标特征向量。

5.根据权利要求4所述的方法，其特征在于，所述图像为所述待追踪视频的第一帧图像，所述方法还包括：

将根据预训练的所述位置向量参数确定的所述特征向量作为所述目标特征向量。

6.根据权利要求4所述的方法，其特征在于，所述对所述图像中所有目标对应的特征向量和所述N个相似特征向量进行去重处理，以得到目标特征向量，包括：

针对所述图像中每一目标对应的特征向量，将所述特征向量与所述N个相似特征向量进行第二相似度计算；

当第二相似度计算结果大于或等于预设相似度时，在所述图像中所有目标对应的特征向量中或所述N个相似特征向量中删除所述第二相似度计算结果对应的特征向量；

将删除处理后所述图像中所有目标对应的特征向量和所述N个相似特征向量中的剩余特征向量作为目标特征向量。

7.根据权利要求4-6任一项所述的方法，其特征在于，所述位置向量参数是通过如下方式训练得到的：

根据初始位置向量参数确定样本图像中目标对应的预测特征向量，以得到所述样本图像对应的预测目标信息，其中所述样本图像预先标注有对应的样本目标信息；

根据所述预测目标信息与所述样本目标信息计算损失函数，并根据所述损失函数的计算结果，调整所述初始位置向量参数。

8.一种视频目标追踪装置，其特征在于，所述装置包括：

获取模块，用于获取待追踪视频；

追踪模块，用于将所述待追踪视频输入目标追踪模型，以得到所述待追踪视频对应的目标追踪结果，所述追踪模块包括：

第一确定子模块，用于针对所述待追踪视频的每一帧图像，确定所述图像对应的目标检测图像中待追踪目标对应的特征向量，所述目标检测图像包括所述待追踪目标；

第二确定子模块，用于将所述图像对应的特征图中的每一特征向量与所述目标检测图像中所述待追踪目标对应的所述特征向量进行第一相似度计算，并根据第一相似度计算结果，在所述特征图的所有特征向量中确定目标特征向量；

第三确定子模块，用于根据所述目标特征向量，确定所述图像中的待追踪目标。