Home

Awesome

Aba-ViTrack

The official implementation of the ICCV 2023 paper Adaptive and Background-Aware Vision Transformer for Real-Time UAV Tracking

<p align="center"> <img width="85%" src="https://github.com/xyyang317/Aba-ViTrack/blob/main/arch.png" alt="Framework"/> </p>

Install the environment

This code has been tested on Ubuntu 18.04, CUDA 10.2. Please install related libraries before running this code:

conda create -n abavitrack python=3.8
conda activate abavitrack
bash install.sh

Model and raw results

The trained model and the raw tracking results are provided in the Baidu Netdisk(code: nen9) or Google Drive.

Run demo

Download the model and put it in checkpoints

python demo.py --initial_bbox 499 421 102 179 

Citation

@InProceedings{Li_2023_ICCV,
    author    = {Li, Shuiwang and Yang, Yangxiang and Zeng, Dan and Wang, Xucheng},
    title     = {Adaptive and Background-Aware Vision Transformer for Real-Time UAV Tracking},
    booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
    month     = {October},
    year      = {2023},
    pages     = {13989-14000}
}