add MOT20 and dancetrack ablation models (#2058)

Author: mikel-brostrom

README.md

@@ -163,11 +163,11 @@ Evaluate a combination of detector, tracking method and ReID model on standard M
 
 ```bash
 # reproduce MOT17 README results
-$ boxmot eval --yolo-model yolox_x_ablation.pt --reid-model lmbn_n_duke.pt --tracking-method boosttrack --source MOT17-ablation --verbose 
+$ boxmot eval --yolo-model yolox_x_MOT17_ablation.pt --reid-model lmbn_n_duke.pt --tracking-method boosttrack --source MOT17-ablation --verbose 
 # MOT20 results
-$ boxmot eval --yolo-model yolox_x_ablation.pt --reid-model lmbn_n_duke.pt --tracking-method boosttrack --source MOT20-ablation --verbose 
+$ boxmot eval --yolo-model yolox_x_MOT20_ablation.pt --reid-model lmbn_n_duke.pt --tracking-method boosttrack --source MOT20-ablation --verbose 
 # Dancetrack results
-$ boxmot eval --yolo-model yolox_x_ablation.pt --reid-model lmbn_n_duke.pt --tracking-method boosttrack --source dancetrack-ablation --verbose 
+$ boxmot eval --yolo-model yolox_x_dancetrack_ablation.pt --reid-model lmbn_n_duke.pt --tracking-method boosttrack --source dancetrack-ablation --verbose 
 # metrics on custom dataset
 $ boxmot eval --yolo-model yolov8n.pt --reid-model osnet_x0_25_msmt17.pt --tracking-method deepocsort  --source ./assets/MOT17-mini/train --verbose
 ```

boxmot/engine/detectors/yolox.py

@@ -1,6 +1,7 @@
 # Mikel Broström 🔥 Yolo Tracking 🧾 AGPL-3.0 license
 
 import cv2
+import fnmatch
 import gdown
 import numpy as np
 import torch
@@ -20,7 +21,9 @@
     "yolox_m.pt": "https://drive.google.com/uc?id=11Zb0NN_Uu7JwUd9e6Nk8o2_EUfxWqsun",
     "yolox_l.pt": "https://drive.google.com/uc?id=1XwfUuCBF4IgWBWK2H7oOhQgEj9Mrb3rz",
     "yolox_x.pt": "https://drive.google.com/uc?id=1P4mY0Yyd3PPTybgZkjMYhFri88nTmJX5",
-    "yolox_x_ablation.pt": "https://drive.google.com/uc?id=1iqhM-6V_r1FpOlOzrdP_Ejshgk0DxOob",
+    "yolox_x_MOT17_ablation.pt": "https://drive.google.com/uc?id=1iqhM-6V_r1FpOlOzrdP_Ejshgk0DxOob",
+    "yolox_x_MOT20_ablation.pt": "https://drive.google.com/uc?id=1H1BxOfinONCSdQKnjGq0XlRxVUo_4M8o",
+    "yolox_x_dancetrack_ablation.pt": "https://drive.google.com/uc?id=1ZKpYmFYCsRdXuOL60NRuc7VXAFYRskXB",
 }
 
 
@@ -132,7 +135,7 @@ def __init__(self, model, device, args):
 
         # download crowdhuman bytetrack models
         if not model.exists() and (
-            model.stem == model_type or model.stem == "yolox_x_ablation"
+            model.stem == model_type or fnmatch.fnmatch(model.stem, "yolox_x_*_ablation")
         ):
             LOGGER.info("Downloading pretrained weights...")
             gdown.download(

boxmot/engine/val.py

@@ -64,7 +64,7 @@ def eval_init(args,
     if args.source in ("MOT17-ablation", "MOT20-ablation", "dancetrack-ablation"):
         cfg = load_dataset_cfg(str(args.source))
         download_eval_data(
-            runs_url=cfg["download"]["dataset_url"],
+            runs_url=cfg["download"]["runs_url"],
             dataset_url=cfg["download"]["dataset_url"],
             dataset_dest=Path(cfg["download"]["dataset_dest"]),
             overwrite=overwrite
@@ -400,7 +400,7 @@ def run_generate_mot_results(opt: argparse.Namespace, evolve_config: dict = None
 
     # Optional GSI
     if getattr(opt, 'gsi', False):
-        from boxmot.utils import gsi
+        from boxmot.postprocessing.gsi import gsi
         gsi(mot_results_folder=exp_dir)