[发明专利]对象跟踪方法、对象跟踪装置和电子设备

说明书

公开了一种对象跟踪方法、对象跟踪装置和电子设备。该对象跟踪方法包括：从视频的当前帧中检测出至少一个目标对象；使用所述至少一个目标对象的关键点确定所述至少一个目标对象的遮挡信息和朝向信息；通过用于重识别的第一神经网络获得所述当前帧的特征图；基于所述至少一个目标对象的遮挡信息、朝向信息和所述特征图获得所述至少一个目标对象的融合跟踪特征；以及，基于所述融合跟踪特征确定所述至少一个目标对象的轨迹。这样，改善了目标对象的跟踪效果。

技术领域

本申请涉及计算机视觉领域，且更为具体地，涉及一种对象跟踪方法，对象跟踪装置和电子设备。

背景技术

多目标多相机跟踪(MTMCT)是计算机视觉中的重要问题，且在公共安全领域中得到广泛应用。MTMCT用于跟踪跨相机的多个目标，这不同于单相机的多对象跟踪(MOT)。多个相机构成的相机网络具有比单个相机更广的视角，因此具有更广泛的应用前景。

MTMCT的主要目的是从由一组相机捕获的视频中提取目标对象的轨迹。近年来，通过采用重识别模型(Re-ID)，MTMCT的跟踪性能得到了显著的增强。但是，在多个相机的场景下，外观特征常常变得不稳定。

另外，在单个相机的场景下，如果视频中存在多个对象，也容易产生诸如目标切换和轨迹断裂等问题。

因此，期望提供改进的对象跟踪方案。

发明内容

为了解决上述技术问题，提出了本申请。本申请的实施例提供了一种对象跟踪方法、对象跟踪装置和电子设备，其结合目标对象的遮挡信息和朝向信息来确定目标对象的轨迹，改善了目标对象的跟踪效果。

根据本申请的一方面，提供了一种对象跟踪方法，包括：从视频的当前帧中检测出至少一个目标对象；使用所述至少一个目标对象的关键点确定所述至少一个目标对象的遮挡信息和朝向信息；通过用于重识别的第一神经网络获得所述当前帧的特征图；基于所述至少一个目标对象的遮挡信息、朝向信息和所述特征图获得所述至少一个目标对象的融合跟踪特征；以及，基于所述融合跟踪特征确定所述至少一个目标对象的轨迹。

根据本申请的另一方面，提供了一种对象跟踪装置，包括：目标检测单元，用于从视频的当前帧中检测出至少一个目标对象；信息确定单元，用于使用所述目标检测单元检测到的所述至少一个目标对象的关键点确定所述至少一个目标对象的遮挡信息和朝向信息；特征图获得单元，用于通过用于重识别的第一神经网络获得所述当前帧的特征图；特征融合单元，用于基于所述信息确定单元所确定的所述至少一个目标对象的遮挡信息、朝向信息和所述特征图获得单元所获得的所述特征图获得所述至少一个目标对象的融合跟踪特征；以及，轨迹确定单元，用于基于所述特征融合单元所获得的所述融合跟踪特征确定所述至少一个目标对象的轨迹。

根据本申请的再一方面，提供了一种电子设备，包括：处理器；以及，存储器，在所述存储器中存储有计算机程序指令，所述计算机程序指令在被所述处理器运行时使得所述处理器执行如上所述的对象跟踪方法。

根据本申请的又一方面，提供了一种计算机可读介质，其上存储有计算机程序指令，所述计算机程序指令在被处理器运行时使得所述处理器执行如上所述的对象跟踪方法。

本申请提供的对象跟踪方法、对象跟踪装置和电子设备通过重识别的神经网络获得特征图，并将特征图与目标对象的遮挡信息和朝向信息相结合，从而能够更好地使用重识别的神经网络所提取的外观特征。这样，使得要跟踪的轨迹特征包含有目标对象的遮挡状态和朝向状态等信息，特别是在多目标场景下能够更准确地确定目标对象的轨迹，改善了目标对象的跟踪效果。

附图说明

通过结合附图对本申请实施例进行更详细的描述，本申请的上述以及其他目的、特征和优势将变得更加明显。附图用来提供对本申请实施例的进一步理解，并且构成说明书的一部分，与本申请实施例一起用于解释本申请，并不构成对本申请的限制。在附图中，相同的参考标号通常代表相同部件或步骤。