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

摘要

本发明涉及计算机技术领域，提供了一种目标跟踪方法、装置、设备及存储介质，所述方法包括：计算获取到的每个原始检测区域和每个预测目标区域之间的马氏距离；将数值最小的马氏距离对应的原始检测目标作为待评估基础目标；若待评估基础目标超过一个，则将预测跟踪目标对应的上一帧视频图像中的原始检测目标作为待处理目标；对每个待处理目标进行深度特征提取，得到每个待处理目标的深度特征向量；将每帧视频图像中的每个待处理目标的深度特征向量和下一帧视频图像中的每个待处理目标的深度特征向量进行匹配，得到每帧视频图像中每个待处理目标对应的匹配结果，并根据匹配结果完成目标跟踪。本发明能够提高目标跟踪的准确度。

主权项

1.一种目标跟踪方法，其特征在于，所述目标跟踪方法包括：获取待处理视频数据，并按照预设的提取方式从所述待处理视频数据中提取预设帧数的视频图像；在预先训练好的目标检测器中，对每帧所述视频图像进行目标检测，得到每帧所述视频图像中的N个原始目标区域，以及每个所述原始目标区域的原始位置信息，并对第一帧所述视频图像中的每个所述原始目标区域包含的原始检测目标设置预设的标识信息，其中，N为正整数；在预先训练好的目标预测器中，对每帧所述视频图像的每个所述原始检测目标进行目标预测，得到每帧所述视频图像中每个所述原始检测目标在下一帧所述视频图像中对应的预测目标区域和所述预测目标区域的预测位置信息，其中，每个所述预测目标区域包含一个预测跟踪目标；在每帧所述视频图像中，根据每个所述原始目标区域的原始位置信息和每个所述预测目标区域的预测位置信息，计算每个所述原始检测区域和每个所述预测目标区域之间的马氏距离；针对每帧所述视频图像的每个所述预测目标区域，从所述预测目标区域对应的马氏距离中选取数值最小的马氏距离，并将该数值最小的马氏距离对应的原始目标区域中的原始检测目标作为所述预测跟踪区域包含的所述预测跟踪目标对应的待评估基础目标；在每帧所述视频图像中，若所述预测跟踪目标仅对应一个所述待评估基础目标，则确认所述待评估基础目标为所述预测跟踪目标对应的上一帧所述视频图像中的所述原始检测目标的跟踪目标，并将所述待评估基础目标的标识信息设置为上一帧所述视频图像中的所述原始检测目标的标识信息；在每帧所述视频图像中，若所述预测跟踪目标对应的所述待评估基础目标超过一个，则将所述预测跟踪目标对应的上一帧所述视频图像中的所述原始检测目标作为待处理目标；对每个所述待处理目标进行深度特征提取，得到每个所述待处理目标的深度特征向量；基于匈牙利匹配算法，将每帧所述视频图像中的每个所述待处理目标的深度特征向量和下一帧所述视频图像中的每个所述待处理目标的深度特征向量进行匹配，得到每帧所述视频图像中每个所述待处理目标对应的匹配结果；针对每帧所述视频图像中的每个所述待处理目标，若当前帧中的所述待处理目标对应的所述匹配结果为匹配成功，则将当前帧中的所述待处理目标在下一帧中的跟踪目标确定为得到该匹配结果的下一帧中的所述待处理目标，并将得到该匹配结果的下一帧中的所述待处理目标的标识信息设置为当前帧中所述待处理目标的标识信息。