算法论文: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