Code Monkey home page Code Monkey logo

globaltrack's Introduction

GlobalTrack

UPDATES:

  • [2020.03.02] Update training scripts to match the settings in the paper (12 epochs on COCO and another 12 epochs on COCO + GOT + LaSOT)!
  • [2020.02.19] Both training and evaluation code are available!
  • [2020.02.19] Initial and pretrained weights are provided!
  • [2020.02.19] A demo tracking video of GlobalTrack is available here!

Official implementation of our AAAI2020 paper: GlobalTrack: A Simple and Strong Baseline for Long-term Tracking. The first tracker with NO cumulative errors.

figure2

Extremely simple tracking process, with NO motion model, NO online learning, NO punishment on position or scale changes, NO scale smoothing and NO trajectory refinement.

Outperforms SPLT (ICCV19), SiamRPN, ATOM and MBMD on TLP benchmark (avg. 13,529 frames per video) by MORE THAN 11% (absolute gain).

Outperforms SPLT, SiamRPN++, ATOM and DaSiamLT on LaSOT benchmark.

Paper on arXiv: 1912.08531.

Demo video: YouTube, YouKu.

Installation

To reproduce our Python environment, you'll need to create a conda environment from environment.yml and compile the Cpp/CUDA extensions (we use CUDA toolkit 9.0):

conda env create -f environment.yml
conda activate GlobalTrack
git clone https://github.com/huanglianghua/GlobalTrack.git
cd _submodules/mmdetection
python setup.py develop

Alternatively, you can also install PyTorch==1.1.0, torchvision, shapely and scipy manually, then compile the Cpp/CUDA extensions by running python setup.py develop under _submodules/mmdetection.

Run Training

(Assuming all datasets are stored in ~/data) Distributed training:

sh tools/dist_train_qg_rcnn.sh

Non-distributed training:

python tools/train_qg_rcnn.py --config configs/qg_rcnn_r50_fpn.py --load_from checkpoints/qg_rcnn_r50_fpn_2x_20181010-443129e1.pth --gpus 1

Before train, you'll need to download the initial weights transferred from FasterRCNN (provided by mmdetection, pretrained on COCO) to start.

Change the arguments in dist_train_qg_rcnn.sh or append them to python tools/train_qg_rcnn.py for your need. See train_qg_rcnn.py for details.

Run Tracking

(Assuming all datasets are stored in ~/data).

python tools/test_global_track.py

Change the parameters, such as cfg_file, ckp_file and evaluators in test_global_track.py for your need.

Pretrained Weights

By defaults, all pretrained weights are saved at checkpoints.

Issues

Please report issues in this repo if you have any problems.

globaltrack's People

Contributors

huanglianghua avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.