GLT-T: Global-Local Transformer Voting for 3D Single Object Tracking in Point Clouds

Authors

  • Jiahao Nie Hangzhou Dianzi University
  • Zhiwei He Hangzhou Dianzi University
  • Yuxiang Yang Hangzhou Dianzi University
  • Mingyu Gao Hangzhou DianZi University
  • Jing Zhang The University of Sydney

DOI:

https://doi.org/10.1609/aaai.v37i2.25287

Keywords:

CV: 3D Computer Vision, CV: Motion & Tracking, CV: Vision for Robotics & Autonomous Driving

Abstract

Current 3D single object tracking methods are typically based on VoteNet, a 3D region proposal network. Despite the success, using a single seed point feature as the cue for offset learning in VoteNet prevents high-quality 3D proposals from being generated. Moreover, seed points with different importance are treated equally in the voting process, aggravating this defect. To address these issues, we propose a novel global-local transformer voting scheme to provide more informative cues and guide the model pay more attention on potential seed points, promoting the generation of high-quality 3D proposals. Technically, a global-local transformer (GLT) module is employed to integrate object- and patch-aware prior into seed point features to effectively form strong feature representation for geometric positions of the seed points, thus providing more robust and accurate cues for offset learning. Subsequently, a simple yet effective training strategy is designed to train the GLT module. We develop an importance prediction branch to learn the potential importance of the seed points and treat the output weights vector as a training constraint term. By incorporating the above components together, we exhibit a superior tracking method GLT-T. Extensive experiments on challenging KITTI and NuScenes benchmarks demonstrate that GLT-T achieves state-of-the-art performance in the 3D single object tracking task. Besides, further ablation studies show the advantages of the proposed global-local transformer voting scheme over the original VoteNet. Code and models will be available at https://github.com/haooozi/GLT-T.

Downloads

Published

2023-06-26

How to Cite

Nie, J., He, Z., Yang, Y., Gao, M., & Zhang, J. (2023). GLT-T: Global-Local Transformer Voting for 3D Single Object Tracking in Point Clouds. Proceedings of the AAAI Conference on Artificial Intelligence, 37(2), 1957-1965. https://doi.org/10.1609/aaai.v37i2.25287

Issue

Section

AAAI Technical Track on Computer Vision II