ViTT: Vision Transformer Tracker

• Published in: Sensors (Basel, Switzerland) Vol. 21; no. 16; p. 5608
• Main Authors: Zhu, Xiaoning, Jia, Yannan, Jian, Sun, Gu, Lize, Pu, Zhang
• Format: Journal Article
• Language: English
• Published: Basel MDPI AG 20.08.2021; MDPI
• Subjects: attention; backbone; Classification; Codec; Computer vision; Deep learning; Kalman filters; MOT; Multiple target tracking; Natural language processing; Neural networks; Occlusion; Task complexity; transformer; Unmanned aerial vehicles
• ISSN: 1424-8220; 1424-8220
• DOI: 10.3390/s21165608

This paper presents a new model for multi-object tracking (MOT) with a transformer. MOT is a spatiotemporal correlation task among interest objects and one of the crucial technologies of multi-unmanned aerial vehicles (Multi-UAV). The transformer is a self-attentional codec architecture that has been successfully used in natural language processing and is emerging in computer vision. This study proposes the Vision Transformer Tracker (ViTT), which uses a transformer encoder as the backbone and takes images directly as input. Compared with convolution networks, it can model global context at every encoder layer from the beginning, which addresses the challenges of occlusion and complex scenarios. The model simultaneously outputs object locations and corresponding appearance embeddings in a shared network through multi-task learning. Our work demonstrates the superiority and effectiveness of transformer-based networks in complex computer vision tasks and paves the way for applying the pure transformer in MOT. We evaluated the proposed model on the MOT16 dataset, achieving 65.7% MOTA, and obtained a competitive result compared with other typical multi-object trackers.