Skip to content

Latest commit

 

History

History
98 lines (78 loc) · 2.48 KB

README.md

File metadata and controls

98 lines (78 loc) · 2.48 KB

MTAtrack

This is the implementation of MTAtrack.
This code is based on DETR: [link].

Tracking demo

d8519df08d38a186718a987f5974f017.mp4

Dependencies

  • PyTorch 1.7.0
  • torchvision 0.8.1
  • cudatoolkit 10.2
  • easydict
  • cython

DataSets

${MTAtrack_ROOT}
 -- data
     -- Lasot
         |-- airplane
         |-- basketball
         |-- bear
         ...
     -- GOT10k
         |-- test
         |-- train
         |-- val
     -- COCO
         |-- annotations
         |-- images
     -- TrackingNet
         |-- TRAIN_0
         |-- TRAIN_1
         ...
         |-- TRAIN_11
         |-- TEST

Trained model

  • Download pre-trained weights into ./checpoints
    [weights]

Code

  • paths setting
python tracking/create_default_local_file.py --workspace_dir . --data_dir ./data --save_dir .
  • TrackingNet validation set
python tracking/test.py MTAtrack_st --dataset trackingnet --threads 16
python lib/test/utils/transform_trackingnet.py --tracker_name MTAtrack_st
  • Lasot validation set
python tracking/test.py MTAtrack_st --dataset lasot --threads 16
python tracking/analysis_results.py  # need to modify tracker configs and names
  • GOT-10k validation set
python tracking/test.py MTAtrack_st --dataset got10k --threads 16
python lib/test/utils/transform_trackingnet.py --tracker_name MTAtrack_st
  • UAV123 validation set
python tracking/test.py MTAtrack_st baseline --dataset uav --threads 16
python tracking/analysis_results.py  # need to modify tracker configs and names
  • OTB100 validation set
python tracking/test.py MTAtrack_st baseline --dataset otb --threads 16
python tracking/analysis_results.py  # need to modify tracker configs and names

Raw results