【多目标跟踪】FairMOT项目测试

参考自:https://blog.csdn.net/sinat_33486980/article/details/105611295
效果好不好不知道，但是肯定简单易懂。

1. 项目地址

https://github.com/ifzhang/FairMOT

2. 效果

FAIRMOT多目标跟踪

3. 依赖

• gcc5.4
• pytorch>=1.2.0
• torchvison==0.4.0
• cuda
• ffmpeg

4. test

1. 下载预训练的模型, 从：https://blog.csdn.net/sinat_33486980/article/details/105611295文章的评论区
   链接：https://pan.baidu.com/s/1mVrEPBl4qe9Uu088kpNiwQ 提取码：6mx6

2. 编译DCNV2，说明用gcc5.4是没有问题的，用gcc6.5会报错，所以报错之后可以检查一下gcc的版本

cd src/lib/models/networks/DCNv2_new sh make.sh

3. 测试

cd src
python demo.py mot --load_model ../models/all_dla34.pth --conf_thres 0.4

4. 结果

• 如果装了ffmpeg，会在$FairMOT/results文件夹下生成result.mp4

FAIRMOT多目标跟踪

• 如果没有装ffmpeg，会在$FairMOT/results/frame文件夹下有结果图片
  

5. 训练

这里用CUHKSYSU数据集做简单的训练。

1. 下载数据集
   链接：https://pan.baidu.com/s/1mwqi13AgSGQfgeb6rYG03Q 提取码：i51f

将数据集解压之后，将CUHK-SYSU改为CUHKSYSU，去掉中间的-

2. 修改训练用数据集配置，data.json
   只用修改root路径
   

3. 在opt.py中修改gpu为0，batchsize为6

4. 训练
   

6. 训练结果测试

。。。