Skip to content

Commit 0c396e7

Browse files
Add dancetrack support (#2052)
1 parent 07c1dcf commit 0c396e7

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -166,7 +166,7 @@ Evaluate a combination of detector, tracking method and ReID model on standard M
166166
$ boxmot eval --yolo-model yolox_x_ablation.pt --reid-model lmbn_n_duke.pt --tracking-method boosttrack --source MOT17-ablation --verbose
167167
# MOT20 results
168168
$ boxmot eval --yolo-model yolox_x_ablation.pt --reid-model lmbn_n_duke.pt --tracking-method boosttrack --source MOT20-ablation --verbose
169-
# Bytetrack results
169+
# Dancetrack results
170170
$ boxmot eval --yolo-model yolox_x_ablation.pt --reid-model lmbn_n_duke.pt --tracking-method boosttrack --source dancetrack-ablation --verbose
171171
# metrics on custom dataset
172172
$ boxmot eval --yolo-model yolov8n.pt --reid-model osnet_x0_25_msmt17.pt --tracking-method deepocsort --source ./assets/MOT17-mini/train --verbose

boxmot/engine/val.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def eval_init(args,
6161
download_trackeval(dest=trackeval_dest, branch=branch, overwrite=overwrite)
6262

6363
# 2) if doing MOT17/20-ablation, pull down the dataset and rewire args.source/split
64-
if args.source in ("MOT17-ablation", "MOT20-ablation", "dancetrack"):
64+
if args.source in ("MOT17-ablation", "MOT20-ablation", "dancetrack-ablation"):
6565
cfg = load_dataset_cfg(str(args.source))
6666
download_eval_data(
6767
runs_url=cfg["download"]["dataset_url"],

0 commit comments

Comments
 (0)