CTTrack

The official implementation of the AAAI 2023 Oral paper [Compact Transformer Tracker with Correlative Masked Modeling]

[Models and Raw results] (Google Driver) [Models and Raw results] (Baidu Driver: LRBW)

News

[ Jan 18, 2023]

• We release Code, models and raw results.

[ Jan 11, 2023]

• Our paper is selected for an oral presentation.

[ Nov 19, 2022]

• CTTrack is accepted to AAAI2023 .

Strong performance

Install the environment

conda create -n cttrack python=3.7
conda activate cttrack
pip install -r requirements.txt

Data Preparation

Put the tracking datasets in ./data. It should look like:

   ${CTTRACK_ROOT}
    -- data
        -- lasot
            |-- airplane
            |-- basketball
            |-- bear
            ...
        -- got10k
            |-- test
            |-- train
            |-- val
        -- trackingnet
            |-- TRAIN_0
            |-- TRAIN_1
            ...
            |-- TRAIN_11
            |-- TEST

Set project paths

Run the following command to set paths for this project

python tracking/create_default_local_file.py --workspace_dir . --data_dir ./data --save_dir .

After running this command, you can also modify paths by editing these two files

lib/train/admin/local.py  # paths about training
lib/test/evaluation/local.py  # paths about testing

Train CTTrack

Train CTTrack-B

python tracking/train.py --script cttrack --config baseline --save_dir . --mode single
python tracking/train.py --script cttrack_online --config baseline --save_dir . --mode single --script_prv cttrak --config_prv baseline  

Train CTTrack-L

python tracking/train.py --script cttrack --config baseline_L --save_dir . --mode single
python tracking/train.py --script cttrack_online --config baseline_L --save_dir . --mode single --script_prv cttrak --config_prv baseline  

Test and evaluate CTTrack on benchmarks

Test CTTrack-B

• OTB2015

  python tracking/test.py cttrack baseline --dataset otb --threads 32

• UAV123

  python tracking/test.py cttrack baseline --dataset uav --threads 32

• LaSOT

  python tracking/test.py cttrack baseline --dataset lasot --threads 32

• GOT10K-test

  python tracking/test.py cttrack baseline --dataset got10k_test --threads 32

• TrackingNet

  python tracking/test.py cttrack baseline --dataset trackingnet --threads 32

  Test CTTrack-L

• OTB2015

  python tracking/test.py cttrack baseline_L --dataset otb --threads 32

• UAV123

  python tracking/test.py cttrack baseline_L --dataset uav --threads 32

• LaSOT

  python tracking/test.py cttrack baseline_L --dataset lasot --threads 32

• GOT10K-test

  python tracking/test.py cttrack baseline_L --dataset got10k_test --threads 32

• TrackingNet

  python tracking/test.py cttrack baseline_L --dataset trackingnet --threads 32

Evaluate CTTrack

LaSOT/GOT10k-test/TrackingNet/OTB100/UAV123

python tracking/analysis_results.py {script}  {config}  {dataset_name}

For example

python tracking/analysis_results.py cttrack baseline trackingnet

VOT2020

Before evaluating "CTTrack" on VOT2020, please install VOT toolkit which is required to evaluate our tracker. To download and install VOT toolkit, you can follow this tutorial. For convenience, you can use our example workspaces of VOT toolkit under external/vot20/ by setting trackers.ini.

cd external/vot20/<workspace_dir>
vot evaluate --workspace . CTTrackPython
# generating analysis results
vot analysis --workspace . CTTrackPython --nocache

Visualize attention maps

Model Zoo and raw results

[Models and Raw results] (Google Driver) [Models and Raw results] (Baidu Driver: RLBW)

Contact

Zikai Song: skyesong@hust.edu.cn

Run Luo: lr_8823@hust.edu.cn

Acknowledgments

• Thanks for PyTracking Library, MixFormer Library and MAE Library, which helps us to quickly implement our ideas.

Citation

If you think this project is helpful, please feel free to leave a star⭐️ and cite our paper:

@InProceedings{Song_2023_AAAI,
   author    = {Song, Zikai and Luo, Run and Yu, Junqing and Chen, Yi-Ping Phoebe and Yang, Wei},
   title     = {Compact Transformer Tracker with Correlative Masked Modeling},
   booktitle = {Proceedings of the AAAI Conference on Artificial Intelligence (AAAI)},
   month     = {February},
   year      = {2023}
}