Joint Object Detection and Multi-Object Tracking with Graph Neural Networks

被引:160
作者
Wang, Yongxin [1 ]
Kitani, Kris [1 ]
Weng, Xinshuo [1 ]
机构
[1] Carnegie Mellon Univ, Robot Inst, Pittsburgh, PA 15213 USA
来源
2021 IEEE INTERNATIONAL CONFERENCE ON ROBOTICS AND AUTOMATION (ICRA 2021) | 2021年
关键词
MULTITARGET;
D O I
10.1109/ICRA48506.2021.9561110
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Object detection and data association are critical components in multi-object tracking (MOT) systems. Despite the fact that the two components are dependent on each other, prior works often design detection and data association modules separately which are trained with separate objectives. As a result, one cannot back-propagate the gradients and optimize the entire MOT system, which leads to sub-optimal performance. To address this issue, recent works simultaneously optimize detection and data association modules under a joint MOT framework, which has shown improved performance in both modules. In this work, we propose a new instance of joint MOT approach based on Graph Neural Networks (GNNs). The key idea is that GNNs can model relations between variable-sized objects in both the spatial and temporal domains, which is essential for learning discriminative features for detection and data association. Through extensive experiments on the MOT15/16/17/20 datasets, we demonstrate the effectiveness of our GNN-based joint MOT approach and show state-of-the-art performance for both detection and MOT tasks.
引用
收藏
页码:13708 / 13715
页数:8
相关论文
共 87 条
  • [1] [Anonymous], 2019, CVPR, DOI DOI 10.1109/CVPR.2019.00300
  • [2] [Anonymous], 2016, ECCV
  • [3] [Anonymous], 2018, ECCV, DOI DOI 10.1007/978-3-030-01240-3_45
  • [4] [Anonymous], 2018, CVPR, DOI DOI 10.1109/ICICS.2018.00068
  • [5] [Anonymous], 2020, CVPR, DOI DOI 10.1109/CVPR42600.2020.00653
  • [6] [Anonymous], 2020, CVPR, DOI DOI 10.1109/CVPR42600.2020.00628
  • [7] [Anonymous], 2018, CVPR, DOI DOI 10.1109/CVPR.2018.00376
  • [8] [Anonymous], 2015, CVPR
  • [9] [Anonymous], 2020, ICML
  • [10] [Anonymous], 2008, CVPR