Awesome
OSTrack
The official implementation for the ECCV 2022 paper Joint Feature Learning and Relation Modeling for Tracking: A One-Stream Framework.
[Models][Raw Results][Training logs]
<!-- [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/joint-feature-learning-and-relation-modeling/visual-object-tracking-on-lasot)](https://paperswithcode.com/sota/visual-object-tracking-on-lasot?p=joint-feature-learning-and-relation-modeling) [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/joint-feature-learning-and-relation-modeling/visual-object-tracking-on-got-10k)](https://paperswithcode.com/sota/visual-object-tracking-on-got-10k?p=joint-feature-learning-and-relation-modeling) [//]: # ([![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/joint-feature-learning-and-relation-modeling/visual-object-tracking-on-trackingnet)](https://paperswithcode.com/sota/visual-object-tracking-on-trackingnet?p=joint-feature-learning-and-relation-modeling)) [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/joint-feature-learning-and-relation-modeling/visual-object-tracking-on-uav123)](https://paperswithcode.com/sota/visual-object-tracking-on-uav123?p=joint-feature-learning-and-relation-modeling) --> <p align="center"> <img width="85%" src="https://github.com/botaoye/OSTrack/blob/main/assets/arch.png" alt="Framework"/> </p>News
[Dec. 12, 2022]
- OSTrack is now available in Modelscope, where you can run demo videos online and conveniently integrate OSTrack into your code.
[Oct. 28, 2022]
- :trophy: We are the winners of VOT-2022 STb(box GT) & RTb challenges.
Highlights
:star2: New One-stream Tracking Framework
OSTrack is a simple, neat, high-performance one-stream tracking framework for joint feature learning and relational modeling based on self-attention operators. Without any additional temporal information, OSTrack achieves SOTA performance on multiple benchmarks. OSTrack can serve as a strong baseline for further research.
Tracker | GOT-10K (AO) | LaSOT (AUC) | TrackingNet (AUC) | UAV123(AUC) |
---|---|---|---|---|
OSTrack-384 | 73.7 | 71.1 | 83.9 | 70.7 |
OSTrack-256 | 71.0 | 69.1 | 83.1 | 68.3 |
:star2: Fast Training
OSTrack-256 can be trained in ~24 hours with 4*V100 (16GB of memory per GPU), which is much faster than recent SOTA transformer-based trackers. The fast training speed comes from:
-
While previous Siamese-style trackers required separate feeding of the template and search region into the backbone at each iteration of training, OSTrack directly combines the template and search region. The tight and highly parallelized structure results in improved training and inference speed.
-
The proposed early candidate elimination (ECE) module significantly reduces memory and time consumption.
-
Pretrained Transformer weights enable faster convergence.
:star2: Good performance-speed trade-off
<p align="center"> <img width="70%" src="https://github.com/botaoye/OSTrack/blob/main/assets/speed_vs_performance.png" alt="speed_vs_performance"/> </p>Install the environment
Option1: Use the Anaconda (CUDA 10.2)
conda create -n ostrack python=3.8
conda activate ostrack
bash install.sh
Option2: Use the Anaconda (CUDA 11.3)
conda env create -f ostrack_cuda113_env.yaml
Option3: Use the docker file
We provide the full docker file here.
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 ./output
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
Data Preparation
Put the tracking datasets in ./data. It should look like this:
${PROJECT_ROOT}
-- data
-- lasot
|-- airplane
|-- basketball
|-- bear
...
-- got10k
|-- test
|-- train
|-- val
-- coco
|-- annotations
|-- images
-- trackingnet
|-- TRAIN_0
|-- TRAIN_1
...
|-- TRAIN_11
|-- TEST
Training
Download pre-trained MAE ViT-Base weights and put it under $PROJECT_ROOT$/pretrained_models
(different pretrained models can also be used, see MAE for more details).
python tracking/train.py --script ostrack --config vitb_256_mae_ce_32x4_ep300 --save_dir ./output --mode multiple --nproc_per_node 4 --use_wandb 1
Replace --config
with the desired model config under experiments/ostrack
. We use wandb to record detailed training logs, in case you don't want to use wandb, set --use_wandb 0
.
Evaluation
Download the model weights from Google Drive
Put the downloaded weights on $PROJECT_ROOT$/output/checkpoints/train/ostrack
Change the corresponding values of lib/test/evaluation/local.py
to the actual benchmark saving paths
Some testing examples:
- LaSOT or other off-line evaluated benchmarks (modify
--dataset
correspondingly)
python tracking/test.py ostrack vitb_384_mae_ce_32x4_ep300 --dataset lasot --threads 16 --num_gpus 4
python tracking/analysis_results.py # need to modify tracker configs and names
- GOT10K-test
python tracking/test.py ostrack vitb_384_mae_ce_32x4_got10k_ep100 --dataset got10k_test --threads 16 --num_gpus 4
python lib/test/utils/transform_got10k.py --tracker_name ostrack --cfg_name vitb_384_mae_ce_32x4_got10k_ep100
- TrackingNet
python tracking/test.py ostrack vitb_384_mae_ce_32x4_ep300 --dataset trackingnet --threads 16 --num_gpus 4
python lib/test/utils/transform_trackingnet.py --tracker_name ostrack --cfg_name vitb_384_mae_ce_32x4_ep300
Visualization or Debug
Visdom is used for visualization.
-
Alive visdom in the server by running
visdom
: -
Simply set
--debug 1
during inference for visualization, e.g.:
python tracking/test.py ostrack vitb_384_mae_ce_32x4_ep300 --dataset vot22 --threads 1 --num_gpus 1 --debug 1
-
Open
http://localhost:8097
in your browser (remember to change the IP address and port according to the actual situation). -
Then you can visualize the candidate elimination process.
Test FLOPs, and Speed
Note: The speeds reported in our paper were tested on a single RTX2080Ti GPU.
# Profiling vitb_256_mae_ce_32x4_ep300
python tracking/profile_model.py --script ostrack --config vitb_256_mae_ce_32x4_ep300
# Profiling vitb_384_mae_ce_32x4_ep300
python tracking/profile_model.py --script ostrack --config vitb_384_mae_ce_32x4_ep300
Acknowledgments
- Thanks for the STARK and PyTracking library, which helps us to quickly implement our ideas.
- We use the implementation of the ViT from the Timm repo.
Citation
If our work is useful for your research, please consider citing:
@inproceedings{ye2022ostrack,
title={Joint Feature Learning and Relation Modeling for Tracking: A One-Stream Framework},
author={Ye, Botao and Chang, Hong and Ma, Bingpeng and Shan, Shiguang and Chen, Xilin},
booktitle={ECCV},
year={2022}
}