jde多目标_GitHub - CnybTseng/JDE: Towards Real-Time Multi-Object Tracking

JDE

Towards Real-Time Multi-Object Tracking

1. 训练步骤

以下操作步骤均以MOT16为例

## 1.1 准备数据集

* 从MOT挑战赛官网下载数据集并解压

cd /data/tseng/dataset/jde

unzip MOT16.zip -d MOT16

* 创建MOT16任务的工作区, 并将MOT格式标注文件转换为需要格式的标注文件

cd JDE

mkdir -p workspace/mot16-2020-5-29

./tools/split_dataset.sh ./workspace/mot16-2020-5-29

此时workspace/mot16-2020-5-29目录下会生成train.txt

1.2 从预训练模型导出参数生成JDE初始模型

从darknet官网下载darknet53预训练模型

wget https://pjreddie.com/media/files/darknet53.conv.74 -P ./workspace

python darknet2pytorch.py -pm ./workspace/mot16-2020-5-29/jde.pth \

--dataset ./workspace/mot16-2020-5-29 -dm ./workspace/darknet53.conv.74 -lbo

此时workspace/mot16-2020-5-29目录下会生成初始模型jde.pth, 其骨干网已初始化为darnet53的参数

1.3 执行训练脚本

cp ./tools/train.sh ./train.sh

根据需要修改, 然后运行训练脚本

./train.sh

2. 测试

运行类似如下命令执行多目标跟踪

python tracker.py --img-path /data/tseng/dataset/jde/MOT16/test/MOT16-03/img1 \

--model workspace/mot16-2020-5-29/checkpoint/jde-ckpt-049.pth