Skip to content

Commit 4150e37

Browse files
MasterBin-IIAUMasterBin-IIAU
committed
Update README
1 parent f1f29c0 commit 4150e37

File tree

2 files changed

+112
-2
lines changed

2 files changed

+112
-2
lines changed

‎README.md

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
[![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/learning-spatio-temporal-transformer-for/visual-object-tracking-on-got-10k)](https://paperswithcode.com/sota/visual-object-tracking-on-got-10k?p=learning-spatio-temporal-transformer-for)
44
[![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/learning-spatio-temporal-transformer-for/visual-object-tracking-on-trackingnet)](https://paperswithcode.com/sota/visual-object-tracking-on-trackingnet?p=learning-spatio-temporal-transformer-for)
55

6-
The official implementation of the paper [**Learning Spatio-Temporal Transformer for Visual Tracking**](https://arxiv.org/abs/2103.17154)
6+
The official implementation of the ICCV2021 paper [**Learning Spatio-Temporal Transformer for Visual Tracking**](https://arxiv.org/abs/2103.17154)
77

88
Hiring research interns for visual transformer projects: houwen.peng@microsoft.com
99

@@ -30,12 +30,21 @@ STARK-ST50 and STARK-ST101 run at **40FPS** and **30FPS** respectively on a Tesl
3030

3131
STARK is implemented purely based on the PyTorch.
3232

33+
## What's new
34+
**July 24, 2021**
35+
- We release an extremely fast version of STARK called **STARK-Lightning** :zap: . It can run at **200~300 FPS** on a RTX TITAN GPU.
36+
Besides, its performance can beat DiMP50, while the model size is even less than that of SiamFC!
37+
- We release a more powerful version of STARK with Swin Transformer as its backbone.
38+
39+
**July 23, 2021**
40+
- STARK is accepted by ICCV2021
41+
3342
## Install the environment
3443
**Option1**: Use the Anaconda
3544
```
3645
conda create -n stark python=3.6
3746
conda activate stark
38-
bash install.sh
47+
bash install_pytorch17.sh
3948
```
4049
**Option2**: Use the docker file
4150

@@ -65,6 +74,7 @@ Put the tracking datasets in ./data. It should look like:
6574
|-- TRAIN_11
6675
|-- TEST
6776
```
77+
## Set project paths
6878
Run the following command to set paths for this project
6979
```
7080
python tracking/create_default_local_file.py --workspace_dir . --data_dir ./data --save_dir .
@@ -129,6 +139,8 @@ python tracking/profile_model.py --script stark_s --config baseline
129139
python tracking/profile_model.py --script stark_st2 --config baseline
130140
# Profiling STARK-ST101 model
131141
python tracking/profile_model.py --script stark_st2 --config baseline_R101
142+
# Profiling STARK-Lightning-X-trt
143+
python tracking/profile_model_lightning_X_trt.py
132144
```
133145

134146
## Model Zoo

‎install_pytorch17.sh

Lines changed: 98 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,98 @@
1+
echo "****************** Installing pytorch ******************"
2+
conda install -y pytorch==1.7.0 torchvision==0.8.1 cudatoolkit=10.2 -c pytorch
3+
4+
echo ""
5+
echo ""
6+
echo "****************** Installing yaml ******************"
7+
pip install PyYAML
8+
9+
echo ""
10+
echo ""
11+
echo "****************** Installing easydict ******************"
12+
pip install easydict
13+
14+
echo ""
15+
echo ""
16+
echo "****************** Installing cython ******************"
17+
pip install cython
18+
19+
echo ""
20+
echo ""
21+
echo "****************** Installing opencv-python ******************"
22+
pip install opencv-python
23+
24+
echo ""
25+
echo ""
26+
echo "****************** Installing pandas ******************"
27+
pip install pandas
28+
29+
echo ""
30+
echo ""
31+
echo "****************** Installing tqdm ******************"
32+
conda install -y tqdm
33+
34+
echo ""
35+
echo ""
36+
echo "****************** Installing coco toolkit ******************"
37+
pip install pycocotools
38+
39+
echo ""
40+
echo ""
41+
echo "****************** Installing jpeg4py python wrapper ******************"
42+
apt-get install libturbojpeg
43+
pip install jpeg4py
44+
45+
echo ""
46+
echo ""
47+
echo "****************** Installing tensorboard ******************"
48+
pip install tb-nightly
49+
50+
echo ""
51+
echo ""
52+
echo "****************** Installing tikzplotlib ******************"
53+
pip install tikzplotlib
54+
55+
echo ""
56+
echo ""
57+
echo "****************** Installing thop tool for FLOPs and Params computing ******************"
58+
pip install --upgrade git+https://github.com/Lyken17/pytorch-OpCounter.git
59+
60+
echo ""
61+
echo ""
62+
echo "****************** Installing colorama ******************"
63+
pip install colorama
64+
65+
echo ""
66+
echo ""
67+
echo "****************** Installing lmdb ******************"
68+
pip install lmdb
69+
70+
echo ""
71+
echo ""
72+
echo "****************** Installing scipy ******************"
73+
pip install scipy
74+
75+
echo ""
76+
echo ""
77+
echo "****************** Installing visdom ******************"
78+
pip install visdom
79+
80+
echo ""
81+
echo ""
82+
echo "****************** Installing vot-toolkit python ******************"
83+
pip install git+https://github.com/votchallenge/vot-toolkit-python
84+
85+
echo ""
86+
echo ""
87+
echo "****************** Installing onnx and onnxruntime-gpu ******************"
88+
pip install onnx onnxruntime-gpu==1.5.1
89+
90+
echo ""
91+
echo ""
92+
echo "****************** Installing timm ******************"
93+
pip install timm==0.3.2
94+
95+
echo ""
96+
echo ""
97+
98+
echo "****************** Installation complete! ******************"

0 commit comments

Comments
 (0)