基于非局部注意力机制的在线多目标跟踪算法

ONLINE MULTI-TARGET TRACKING ALGORITHM BASED ON NON-LOCAL ATTENTION MECHANISM

  • 摘要: 针对多目标跟踪任务在人群拥挤场景存在目标漏检、遮挡等问题,在CenterTrack框架基础上引入非局部注意力机制以捕捉多个目标之间、目标与场景之间的非局部依赖关系,提出基于空间非局部注意力残差块的跟踪模型;并进一步扩展到时空域,建立基于时空关系非局部注意力模块的跟踪模型,同时实现检测和跟踪任务。在MOT17、MOT16、2DMOT15三个数据集的实验结果表明,提出的两种在线跟踪模型较CenterTrack算法有明显提升,且在MOT17中MOTA(Multiple Object Tracking Accuracy)指标达到了目前较为先进的水平,为62.4%和62.5%,验证了该算法的有效性。

     

    Abstract: Aimed at the problems of missing detection and blocking of targets in crowded scenes in multi-target tracking, the non-local attention mechanism is introduced on the basis of the CenterTrack framework to capture the non-local dependencies between multiple targets and scenes. A tracking model based on spatial non-local attention residual block was proposed. We further extended it to the spatiotemporal domain, established a tracking model based on the spatiotemporal relationship non-local attention module, which achieved detection and tracking tasks simultaneously. The experimental results on the MOT17, MOT16 and 2DMOT15 dataset show that the two proposed online tracking models are significantly improved compared with CenterTrack, and reach the current relatively excellent level of 62.4% and 62.5% in MOTA on MOT17 challenge, which verifies the effectiveness of the proposed algorithms.

     

/

返回文章
返回