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

说明书

本公开提供了一种目标追踪方法及装置、电子设备和存储介质，涉及视频处理技术领域，尤其涉及目标追踪技术领域。实现方案为：获取目标视频的目标视频帧中对应于追踪目标的检测框；获取所述目标视频的当前待匹配视频帧中的一个或多个检测目标以及包围每一个检测目标的检测框；确定所述追踪目标的交并比阈值；针对所述追踪目标，计算所述目标视频帧中所述追踪目标对应的检测框和所述当前待匹配视频帧中每一个检测目标的检测框之间的交并比值；针对所述追踪目标，至少基于所述追踪目标的交并比阈值和对应于所述当前待匹配视频帧的一个或多个交并比值，确定所述追踪目标是否与所述当前待匹配视频帧中的一个或多个检测目标中的一个检测目标匹配。

技术领域

本公开涉及人工智能技术领域，尤其涉及目标追踪技术领域，具体涉及一种目标追踪方法及装置、电子设备、计算机可读存储介质和计算机程序产品。

背景技术

目标追踪技术属于机器视觉领域，广泛应用于智慧安防、自动驾驶、行为分析等视频类产品中。

基于交并比(IOU)值的追踪器(IOU-tracker)，通过对追踪目标在前后视频帧中的位置关联实现目标追踪，具体通过计算追踪目标和检测器检测的检测目标之间位置的交并比(IOU)值，并基于交并比策略完成目标追踪。由于其无需使用度量的方法(如相似度度量等)，实现过程简单、速度快，基于TBD(tracking-by-dection)而不需要图像信息，因此广受关注。

在此部分中描述的方法不一定是之前已经设想到或采用的方法。除非另有指明，否则不应假定此部分中描述的任何方法仅因其包括在此部分中就被认为是现有技术。类似地，除非另有指明，否则此部分中提及的问题不应认为在任何现有技术中已被公认。

发明内容

本公开提供了一种目标追踪方法及装置、电子设备、计算机可读存储介质和计算机程序产品。

根据本公开的一方面，提供了一种目标追踪方法，包括：获取目标视频的目标视频帧中对应于一个或多个追踪目标中的每一个追踪目标的检测框；获取所述目标视频的当前待匹配视频帧中的一个或多个检测目标以及包围每一个检测目标的检测框；分别确定所述一个或多个追踪目标中每一个追踪目标的交并比阈值；针对所述一个或多个追踪目标中的每一个追踪目标，计算所述目标视频帧中所述追踪目标对应的检测框和所述当前待匹配视频帧中每一个检测目标的检测框之间的交并比值，以得到所述追踪目标对应于所述当前待匹配视频帧的一个或多个交并比值；以及，针对所述一个或多个追踪目标中的每一个追踪目标，至少基于所述追踪目标的交并比阈值和对应于所述当前待匹配视频帧的一个或多个交并比值，确定所述追踪目标是否与所述当前待匹配视频帧中的一个或多个检测目标中的一个检测目标匹配。

根据本公开的另一方面，还提供一种目标追踪装置，包括：获取单元，被配置用于获取目标视频的目标视频帧中对应于一个或多个追踪目标中的每一个追踪目标的检测框；检测单元，被配置用于获取所述目标视频的当前待匹配视频帧中的一个或多个检测目标以及包围每一个检测目标的检测框；第一确定单元，被配置用于分别确定所述一个或多个追踪目标中每一个追踪目标的交并比阈值；计算单元，被配置用于针对所述一个或多个追踪目标中的每一个追踪目标，计算所述目标视频帧中所述追踪目标对应的检测框和所述当前待匹配视频帧中每一个检测目标的检测框之间的交并比值，以得到所述追踪目标对应于所述当前待匹配视频帧的一个或多个交并比值；以及第二确定单元，被配置用于针对所述一个或多个追踪目标中的每一个追踪目标，至少基于所述追踪目标的交并比阈值和对应于所述当前待匹配视频帧的一个或多个交并比值，确定所述追踪目标是否与所述当前待匹配视频帧中的一个或多个检测目标中的一个检测目标匹配。

根据本公开的另一方面，还提供一种电子设备，包括：至少一个处理器；以及与所述至少一个处理器通信连接的存储器；其中所述存储器存储有计算机程序，所述计算机程序在被所述至少一个处理器执行时实现根据上述的方法。

根据本公开的另一方面，还提供一种存储有计算机程序的非瞬时计算机可读存储介质，其中，所述计算机程序在被处理器执行时实现根据上述的方法。