从github下载DanceTrack项目(不包含数据集,数据集需要单独下载)
git clone https://github.com/DanceTrack/DanceTrack.git
修改convert_dance_to_coco.py中数据集的位置,然后运行,将数据集格式转化为coco格式
将DanceTrack中ByteTrack中的相关文件拷贝到ByteTrack项目中
DanceTrack中ByteTrack目录下的所有文件都要替换到ByteTrack中
将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改为' '即可