Learning a Proposal Classifier for Multiple Object Tracking

被引:80
作者
Dai, Peng [1 ]
Weng, Renliang [2 ]
Choi, Wongun [2 ]
Zhang, Changshui [1 ]
He, Zhangping [2 ]
Ding, Wei [1 ]
机构
[1] Tsinghua Univ, Beijing, Peoples R China
[2] Aibee Inc, Beijing, Peoples R China
来源
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021 | 2021年
关键词
D O I
10.1109/CVPR46437.2021.00247
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The recent trend in multiple object tracking (MOT) is heading towards leveraging deep learning to boost the tracking performance. However, it is not trivial to solve the data-association problem in an end-to-end fashion. In this paper, we propose a novel proposal-based learnable framework, which models MOT as a proposal generation, proposal scoring and trajectory inference paradigm on an affinity graph. This framework is similar to the two-stage object detector Faster RCNN, and can solve the MOT problem in a data-driven way. For proposal generation, we propose an iterative graph clustering method to reduce the computational cost while maintaining the quality of the generated proposals. For proposal scoring, we deploy a trainable graph-convolutional-network (GCN) to learn the structural patterns of the generated proposals and rank them according to the estimated quality scores. For trajectory inference, a simple deoverlapping strategy is adopted to generate tracking output while complying with the constraints that no detection can be assigned to more than one track. We experimentally demonstrate that the proposed method achieves a clear performance improvement in both MOTA and IDF1 with respect to previous state-of-the-art on two public benchmarks.
引用
收藏
页码:2443 / 2452
页数:10
相关论文
共 61 条
  • [1] Ahyja R.K., 1993, Network flows: theory, algorithms, and applications
  • [2] [Anonymous], 2017, CVPR, DOI DOI 10.1109/CVPR.2017.394
  • [3] [Anonymous], 2016, CVPR WORKSH, DOI DOI 10.1109/CVPRW.2016.59
  • [4] Bai S., 2018, ARXIV180301271, DOI [10.48550/arXiv.1803.01271, DOI 10.48550/ARXIV.1803.01271]
  • [5] Multiple Object Tracking Using K-Shortest Paths Optimization
    Berclaz, Jerome
    Fleuret, Francois
    Tueretken, Engin
    Fua, Pascal
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2011, 33 (09) : 1806 - 1819
  • [6] Tracking without bells and whistles
    Bergmann, Philipp
    Meinhardt, Tim
    Leal-Taixe, Laura
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 941 - 951
  • [7] Evaluating Multiple Object Tracking Performance: The CLEAR MOT Metrics
    Bernardin, Keni
    Stiefelhagen, Rainer
    [J]. EURASIP JOURNAL ON IMAGE AND VIDEO PROCESSING, 2008, 2008 (1)
  • [8] Bewley A, 2016, IEEE IMAGE PROC, P3464, DOI 10.1109/ICIP.2016.7533003
  • [9] Brendel W, 2011, PROC CVPR IEEE, P1273, DOI 10.1109/CVPR.2011.5995395
  • [10] Multi-target Tracking by Lagrangian Relaxation to Min-Cost Network Flow
    Butt, Asad A.
    Collins, Robert T.
    [J]. 2013 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2013, : 1846 - 1853