在ByteTrack中使用DanceTrack数据集
从github下载DanceTrack项目(不包含数据集,数据集需要单独下载)
git clone https://github.com/DanceTrack/DanceTrack.git

修改convert_dance_to_coco.py中数据集的位置,然后运行,将数据集格式转化为coco格式

将DanceTrack中ByteTrack中的相关文件拷贝到ByteTrack项目中
将DanceTrack项目中的ByteTrack文件夹中所有文件移动到ByteTrack项目中对应的文件夹,原本的ByteTrack文件新建文件夹进行保存,便于之后运行非DanceTrack数据集,将tools/track.py重命名为track_dance.py,然后再移动过去,避免覆盖原来的track.py文件

将DanceTrack中的三个SEQMAP文件拷贝到已经下载好的DanceTrack数据集下

在ByteTrack中,测试DanceTrack验证集,测试结果保存在YOLOX_outputs/yolox_x/track_results
python tools/track_dance.py -f exps/example/dancetrack/yolox_x.py -c /home/czh/work/MOT/datasets/DanceTrack/bytetrack_models/bytetrack_model.pth.tar -b 1 -d 1 --fp16 --fuse

在 DanceTrack中的TrackEval上进行评估
python3 TrackEval/scripts/run_mot_challenge.py --SPLIT_TO_EVAL val --METRICS HOTA CLEAR Identity --GT_FOLDER /home/czh/work/MOT/datasets/DanceTrack/val --SEQMAP_FILE /home/czh/work/MOT/datasets/DanceTrack/val_seqmap.txt --SKIP_SPLIT_FOL True --TRACKERS_TO_EVAL '' --TRACKER_SUB_FOLDER '' --USE_PARALLEL True --NUM_PARALLEL_CORES 1 --PLOT_CURVES False --TRACKERS_FOLDER /home/czh/work/MOT/ByteTrack/YOLOX_outputs/yolox_x/track_results
其中,--SPLIT_TO_EVAL选定需要评估训练集还是验证集 ,值可以取'train', 'val',对应对应数据集的存放文件夹名称;
--METRICS是选择需要评估的指标;
--GT_FOLDER是真实值存放的路径;
--SEQMAP_FILE数据集提供的数据映射文件;
--SKIP_SPLIT_FOL True跳过中间的 "benchmark-split "文件夹;
--PLOT_CURVES是否绘制结果曲线;
--TRACKERS_FOLDER追踪器运行结果存放的文件夹。
官方实验结果:

自己的评估结果:



如何把从(https://github.com/JonathonLuiten/TrackEval)下载的TrackEval进行修改用来评估DanceTrack验证集,不使用DanceTrack项目自带的?

将原本'SEQMAP_FILE'对应的None改为' '即可

浙公网安备 33010602011771号