Efficient Single-Object Tracker Based on Local-Global Feature Fusion
Xiaoyu Ni, Liang Yuan, Kai Lv
Abstract
Since Vision Transformers (ViTs) are introduced into computer vision, they have developed rapidly in a variety of visual tasks. Recently, they have been gradually applied to visual tracking. The Transformer can adaptively capture the global similarity comparisons of target objects and search regions, which has achieved competitive performance results. However, Transformer architectures often require a large amount of training data and computing resources, and lack prior knowledge of inductive biases that existed in images. The advantages of convolutional neural networks (CNNs) in extracting local similarities are not fully exploited. To resolve these problems, we propose a lightweight tracking architecture, combining CNN and Transformer in the feature fusion stage. Specifically, Local-Global Feature Interaction (LGFI) module and Feature Cross-Fusion (FCF) module are the key components in our approach. In the LGFI module, the proposed method includes a Transformer global information network and a Transformer-like CNN local information network for simultaneous global scope dependency establishment and local feature similarity enhancement, then aggregates their feature results together. In the FCF module, the proposed method includes a multi-head cross-attention and a convolutional feedforward network for feature fusion of templates and search regions. Finally, we use the classification and regression head to predict the exact location of the target. Extensive experiments demonstrate that, our method achieves better tracking performance than the baseline method, when both methods are trained with fewer data. Meanwhile, without any extra training data, the proposed method also obtains comparable results with other state-of-the-art trackers on six challenging benchmarks, including GOT-10k, LaSOT, TrackingNet, OTB100, UAV123, and NFS. Furthermore, our model is lightweight compared with the baseline method, with fewer parameters and lower FLOPs, while running at real-time speed.