刚开始学习多目标跟踪的小白,继上次复现了sort之后,这次准备复现一下deepsort。
deepsort相较于sort增加了ReID类的特征提取和马氏距离计算相融合的技术,明显改善了IDswitch的次数。
论文下载:
代码地址:https://github.com/nwojke/deep_sort
代码直接解压缩即可。
需要的环境(本人使用的环境):
python3.6、scikit-learn==0.22.2、tensorflow1.14.0、numpy。
需要下载的文件:
- MOT16数据集。官网下载地址:https://motchallenge.net/data/MOT16/ (但是官网下载可能会非常慢)。 百度网盘下载地址(百度网盘下载的稍微快些):https://pan.baidu.com/s/1fIqeqGizP6RelWGezDwZuQ 提取码:8aaq
- 下载预训练好的目标检测文件和特征训练文件。下载地址:https://drive.google.com/drive/folders/18fKzfqnqhqW3s9zwsCbnVJ5XF2JFeqMp(需要使用加速器进)百度网盘下载地址:https://pan.baidu.com/s/13Y4mJzbfyi365rsQOtSlVQ 提取码:yrd4
- 下载后解压到相应位置,文件放置如图:
- 创建tmp文件夹,在tmp文件夹下创建hypotheses.txt文件,并修改deep_sort_app.py文件中第234行,将 /tmp/hypotheses.txt修改为 ./tmp/hypotheses.txt。
- 在控制台输入运行命令:D:\MOTpro\deep_sort-master\deep_sort-master>python deep_sort_app.py --sequence_dir=./MOT16/test/MOT16-06 --detection_file=./resources/detections/MOT16_POI_test/MOT16-06.npy --min_confidence=0.3 --nn_budget=100 --display=True
- 运行成功。在hypotheses.txt生成运行结果。
本人是刚要学习多目标跟踪的小白,欢迎大家指导交流。