Tracking Objects as Points(CenterTrack)多目标跟踪算法运行记录

算法论文：https://arxiv.org/abs/2004.01177
如果打不开就把前面的https://arxiv.org/abs/换成http://xxx.itp.ac.cn/abs/，秒开
代码地址：https://github.com/xingyizhou/CenterTrack
我的环境：ubuntu18.04 python3.6 pytorch1.1

创建我的环境(为了不改动anaconda的主环境，万一把主环境搞崩了呢)

conda create --name CenterTrack python=3.6

激活环境

source activate CenterTrack

安装pytorch，//这里建议手动离线安装pytorch，在线安装容易失败，我离线安装的版本是pytorch1.1，torchvision也要安装与pytorch对应的版本

conda install pytorch torchvision -c pytorch

安装COCOAPI

pip install cython; pip install -U 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'

下载代码并把命令窗口cd到代码工程中，//例如我的是

cd /home/pc/CenterTrack/CenterTrack-master

安装必要的包

pip install -r requirements.txt

编译DCNv2

cd /src/lib/model/networks/

git clone https://github.com/CharlesShang/DCNv2/

cd DCNv2

./make.sh

在https://github.com/xingyizhou/CenterTrack/blob/master/readme/MODEL_ZOO.md 下载预训练模型并放入对应的文件夹，

我下载的是mot17_fulltrain.pth,在工程主目录新建一个models文件夹,把下载的模型放进去。作者给的是google网盘需要，我这里下载好了，放在百度网盘里：
链接：https://pan.baidu.com/s/13-MJCVPtB7HxTOYNursCHg
提取码：t5lc

以上准备工作完成后，把命令窗口cd到

cd /home/pc/CenterTrack/CenterTrack-master/src

然后运行，这里需要自己更改图片序列的文件夹地址或者视频地址

python demo.py tracking --load_model ../models/mot17_fulltrain.pth --demo /path/to/your/image/or/folder/or/video

以上配置如果没错的话，运行成功，但是你会发现输出的图像中并没有目标框，一个都没有！

所以要改一个地方在/CenterTrack-master/src/lib中的opts.py中修改

#396行
#train_dataset = default_dataset_info[opt.task] \
      #if opt.task in default_dataset_info else 'coco'
 train_dataset = opt.dataset

然后运行命令行

python demo.py tracking --load_model ../models/mot17_fulltrain.pth --demo /path/to/your/image/or/folder/or/video  --dataset mot

即可。

有问题欢迎交流，我会及时回复。