1. 项目地址

https://github.com/ifzhang/FairMOT

2. 效果

 

FAIRMOT多目标跟踪

 

3. 依赖

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

4. test

  1. 下载预训练的模型, 从：javascript:void(0)文章的评论区
    链接：https://pan.baidu.com/s/1mVrEPBl4qe9Uu088kpNiwQ 提取码：6mx6

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

cd src/lib/models/networks/DCNv2_new sh make.sh
  1. 测试
cd src
python demo.py mot --load_model ../models/all_dla34.pth --conf_thres 0.4
  1. 结果
  • 如果装了ffmpeg，会在$FairMOT/results文件夹下生成result.mp4

 

FAIRMOT多目标跟踪

 

  • 如果没有装ffmpeg，会在$FairMOT/results/frame文件夹下有结果图片
    【多目标跟踪】FairMOT项目测试_json

5. 训练

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

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

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

  1. 修改训练用数据集配置，data.json
    只用修改root路径
    【多目标跟踪】FairMOT项目测试_数据集_02

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

  3. 训练
    【多目标跟踪】FairMOT项目测试_json_03

6. 训练结果测试

。。。