LightTrack: Finding Lightweight Neural Networks for Object Tracking via One-Shot Architecture Search
The official implementation of the paper
LightTrack: Finding Lightweight Neural Networks for Object Tracking via One-Shot Architecture Search
Hiring research interns for visual transformer projects: houwen.peng@microsoft.com
Environment Installation
cd lighttrack
conda create -n lighttrack python=3.6
conda activate lighttrack
bash install.sh
Data Preparation
- Tracking Benchmarks
Please put VOT2019 dataset under $LightTrack/dataset
. The prepared data should look like:
$LighTrack/dataset/VOT2019.json
$LighTrack/dataset/VOT2019/agility
$LighTrack/dataset/VOT2019/ants1
...
$LighTrack/dataset/VOT2019/list.txt
Test and evaluation
Test LightTrack-Mobile on VOT2019
bash tracking/reproduce_vot2019.sh
Flops, Params, and Speed
Compute the flops and params of our LightTrack-Mobile. The flops counter we use is pytorch-OpCounter
python tracking/FLOPs_Params.py
Test the running speed of our LightTrack-Mobile
python tracking/Speed.py