# PyTracking
A general python library for visual tracking algorithms.
## Table of Contents
* [Running a tracker](#running-a-tracker)
* [Overview](#overview)
* [Trackers](#trackers)
* [LWL](#LWL)
* [KYS](#KYS)
* [DiMP](#DiMP)
* [ATOM](#ATOM)
* [ECO](#ECO)
* [Analysis](#analysis)
* [Libs](#libs)
* [Visdom](#visdom)
* [VOT Integration](#vot-integration)
* [Integrating a new tracker](#integrating-a-new-tracker)
## Running a tracker
The installation script will automatically generate a local configuration file "evaluation/local.py". In case the file was not generated, run ```evaluation.environment.create_default_local_file()``` to generate it. Next, set the paths to the datasets you want
to use for evaluations. You can also change the path to the networks folder, and the path to the results folder, if you do not want to use the default paths. If all the dependencies have been correctly installed, you are set to run the trackers.
The toolkit provides many ways to run a tracker.
**Run the tracker on webcam feed**
This is done using the run_webcam script. The arguments are the name of the tracker, and the name of the parameter file. You can select the object to track by drawing a bounding box. **Note:** It is possible to select multiple targets to track!
```bash
python run_webcam.py tracker_name parameter_name
```
**Run the tracker on some dataset sequence**
This is done using the run_tracker script.
```bash
python run_tracker.py tracker_name parameter_name --dataset_name dataset_name --sequence sequence --debug debug --threads threads
```
Here, the dataset_name is the name of the dataset used for evaluation, e.g. ```otb```. See [evaluation.datasets.py](evaluation/datasets.py) for the list of datasets which are supported. The sequence can either be an integer denoting the index of the sequence in the dataset, or the name of the sequence, e.g. ```'Soccer'```.
The ```debug``` parameter can be used to control the level of debug visualizations. ```threads``` parameter can be used to run on multiple threads.
**Run the tracker on a set of datasets**
This is done using the run_experiment script. To use this, first you need to create an experiment setting file in ```pytracking/experiments```. See [myexperiments.py](experiments/myexperiments.py) for reference.
```bash
python run_experiment.py experiment_module experiment_name --dataset_name dataset_name --sequence sequence --debug debug --threads threads
```
Here, ```experiment_module``` is the name of the experiment setting file, e.g. ```myexperiments``` , and ``` experiment_name``` is the name of the experiment setting, e.g. ``` atom_nfs_uav``` .
**Run the tracker on a video file**
This is done using the run_video script.
```bash
python run_video.py experiment_module experiment_name videofile --optional_box optional_box --debug debug
```
Here, ```videofile``` is the path to the video file. You can either draw the box by hand or provide it directly in the ```optional_box``` argument.
## Overview
The tookit consists of the following sub-modules.
- [analysis](analysis): Contains scripts to analyse tracking performance, e.g. obtain success plots, compute AUC score. It also contains a [script](analysis/playback_results.py) to playback saved results for debugging.
- [evaluation](evaluation): Contains the necessary scripts for running a tracker on a dataset. It also contains integration of a number of standard tracking and video object segmentation datasets, namely [OTB-100](http://cvlab.hanyang.ac.kr/tracker_benchmark/index.html), [NFS](http://ci2cv.net/nfs/index.html),
[UAV123](https://ivul.kaust.edu.sa/Pages/pub-benchmark-simulator-uav.aspx), [Temple128](http://www.dabi.temple.edu/~hbling/data/TColor-128/TColor-128.html), [TrackingNet](https://tracking-net.org/), [GOT-10k](http://got-10k.aitestunion.com/), [LaSOT](https://cis.temple.edu/lasot/), [VOT](http://www.votchallenge.net), [Temple Color 128](http://www.dabi.temple.edu/~hbling/data/TColor-128/TColor-128.html), [DAVIS](https://davischallenge.org), and [YouTube-VOS](https://youtube-vos.org).
- [experiments](experiments): The experiment setting files must be stored here,
- [features](features): Contains tools for feature extraction, data augmentation and wrapping networks.
- [libs](libs): Includes libraries for optimization, dcf, etc.
- [notebooks](notebooks) Jupyter notebooks to analyze tracker performance.
- [parameter](parameter): Contains the parameter settings for different trackers.
- [tracker](tracker): Contains the implementations of different trackers.
- [util_scripts](util_scripts): Some util scripts for e.g. generating packed results for evaluation on GOT-10k and TrackingNet evaluation servers, downloading pre-computed results.
- [utils](utils): Some util functions.
- [VOT](VOT): VOT Integration.
## Trackers
The toolkit contains the implementation of the following trackers.
### LWL
The official implementation for the LWL tracker ([paper](https://arxiv.org/pdf/2003.11540.pdf)).
The tracker implementation file can be found at [tracker.lwl](tracker/lwl).
##### Parameter Files
Two parameter settings are provided. These can be used to reproduce the results or as a starting point for your exploration.
* **[lwl_ytvos](parameter/lwl/lwl_ytvos.py)**: The default parameter setting with ResNet-50 backbone which was used to generate YouTubeVOS results.
* **[lwl_boxinit](parameter/lwl/lwl_boxinit.py)**: The parameters settings used to generate results with bounding box initialization on YouTubeVOS and DAVIS datasets.
### KYS
The official implementation for the KYS tracker ([paper](https://arxiv.org/pdf/2003.11014.pdf)).
The tracker implementation file can be found at [tracker.kys](tracker/kys).
##### Parameter Files
* **[default](parameter/kys/default.py)**: The default parameter setting with ResNet-50 backbone which was used to produce all results in the paper, except on VOT and LaSOT.
* **[default_vot](parameter/kys/default_vot.py)**: The parameters settings used to generate the VOT2018 results in the paper.
### DiMP
The official implementation for the DiMP tracker ([paper](https://arxiv.org/abs/1904.07220)) and PrDiMP tracker ([paper](https://arxiv.org/abs/2003.12565)).
The tracker implementation file can be found at [tracker.dimp](tracker/dimp).
##### Parameter Files
* **[dimp18](parameter/dimp/dimp18.py)**: The default parameter setting with ResNet-18 backbone which was used to produce all DiMP-18 results in the paper, except on VOT.
* **[dimp18_vot](parameter/dimp/dimp18_vot18.py)**: The parameters settings used to generate the DiMP-18 VOT2018 results in the paper.
* **[dimp50](parameter/dimp/dimp50.py)**: The default parameter setting with ResNet-50 backbone which was used to produce all DiMP-50 results in the paper, except on VOT.
* **[dimp50_vot](parameter/dimp/dimp50_vot18.py)**: The parameters settings used to generate the DiMP-50 VOT2018 results in the paper.
* **[prdimp18](parameter/dimp/prdimp18.py)**: The default parameter setting with ResNet-18 backbone which was used to produce all PrDiMP-18 results in the paper, except on VOT.
* **[prdimp50](parameter/dimp/prdimp50.py)**: The default parameter setting with ResNet-50 backbone which was used to produce all PrDiMP-50 results in the paper, except on VOT.
* **[super_dimp](parameter/dimp/super_dimp.py)**: Combines the bounding-box regressor of PrDiMP with the standard DiMP classifier and better training and inference settings.
The difference between the VOT and the non-VOT settings stems from the fact that the VOT protocol measures robustness in a very different manner compared to other benchmarks. In most benchmarks, it is highly important to be able to robustly *redetect* the target after e.g. an occlusion or brief target loss. On the other hand, in VOT the tracker is reset if the prediction does not overlap with the target on a *single* frame. This is then counted as a tra
没有合适的资源?快使用搜索试试~ 我知道了~
资源推荐
资源详情
资源评论
收起资源包目录
算法部署-使用TensorRT部署AlphaRefine目标跟踪算法-优质算法部署项目实战.zip (652个子文件)
region.c 509KB
region.c 23KB
prroi_pooling_gpu.c 4KB
setup.cfg 40B
roi_align_cuda.cpp 5KB
roi_align_kernel_c.cu 20KB
prroi_pooling_gpu_impl.cu 17KB
prroi_pooling_gpu_impl.cu 17KB
roi_align_kernel.cu 11KB
prroi_pooling_gpu_impl.cuh 2KB
prroi_pooling_gpu_impl.cuh 2KB
.gitignore 1KB
.gitignore 1KB
.gitignore 110B
.gitignore 64B
.gitignore 20B
.gitignore 0B
buffer.h 4KB
region.h 3KB
prroi_pooling_gpu.h 865B
MANIFEST.in 71B
trackers.ini 399B
quick_examples.ipynb 324KB
analyze_results.ipynb 7KB
sample_batch_run.jpg 198KB
UAV123.json 5.34MB
lasot.json 28KB
uav123.json 18KB
LICENSE 11KB
LICENSE 1KB
LICENSE 1KB
tracker_DiMP.m 1KB
README.md 13KB
README.md 9KB
README.md 8KB
README.md 7KB
README.md 6KB
README.md 4KB
MODEL_ZOO.md 4KB
TRAIN.md 3KB
README.md 2KB
INSTALL.md 895B
README.md 1B
README.md 1B
README.md 1B
README.md 1B
README.md 1B
kys_overview.png 242KB
prroi_visualization.png 186KB
visdom.png 137KB
atom_overview.png 121KB
lwtl_overview.png 109KB
dimp_overview.png 98KB
NFS.png 32KB
LaSOT.png 31KB
UAV123.png 30KB
OTB-100.png 28KB
cow.png 14KB
c_region.pxd 1KB
processing.py 49KB
kys.py 40KB
dimp.py 40KB
atom.py 39KB
RT_MDNet.py 30KB
run_siamrpn_RF.py 30KB
tracker.py 29KB
sampler.py 28KB
lwl.py 27KB
uavdataset.py 27KB
tracker.py 26KB
tpldataset.py 26KB
utils.py 24KB
nfsdataset.py 23KB
optimizer.py 23KB
processing_utils.py 23KB
vot.py 22KB
plot_results.py 22KB
optimization.py 21KB
otbdataset.py 21KB
tracking.py 19KB
dimpnet.py 19KB
trax.py 19KB
eco.py 17KB
run_dimp_RF.py 17KB
run_dimp_RF3.py 17KB
run_dimp_RF2.py 17KB
model.py 17KB
transforms.py 16KB
processing_utils_SE.py 16KB
processing_utils_SE.py 16KB
visdom.py 16KB
vos_base.py 16KB
processing.py 16KB
run_dimpsuper_RF.py 15KB
got10k.py 15KB
measures.py 15KB
lwl_box_net.py 14KB
processing_utils.py 13KB
cli.py 13KB
lasotdataset.py 13KB
共 652 条
- 1
- 2
- 3
- 4
- 5
- 6
- 7
资源评论
极智视界
- 粉丝: 2w+
- 资源: 1459
下载权益
C知道特权
VIP文章
课程特权
开通VIP
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功