new configs folder

boxmot/configs/defaults/__init__.py

@@ -0,0 +1 @@
+# Mikel Broström 🔥 Yolo Tracking 🧾 AGPL-3.0 license

boxmot/configs/search/__init__.py

@@ -0,0 +1 @@
+# Mikel Broström 🔥 Yolo Tracking 🧾 AGPL-3.0 license

boxmot/configs/search/botsort.yaml

@@ -0,0 +1,37 @@
+# botsort_search_space.yaml
+
+track_high_thresh:
+  type: uniform
+  range: [0.3, 0.7]
+
+track_low_thresh:
+  type: uniform
+  range: [0.1, 0.3]
+
+new_track_thresh:
+  type: uniform
+  range: [0.1, 0.8]
+
+track_buffer:
+  type: randint
+  range: [20, 81]
+
+match_thresh:
+  type: uniform
+  range: [0.1, 0.9]
+
+proximity_thresh:
+  type: uniform
+  range: [0.25, 0.75]
+
+appearance_thresh:
+  type: uniform
+  range: [0.1, 0.8]
+
+cmc_method:
+  type: choice
+  options: ['sparseOptFlow']
+
+frame_rate:
+  type: choice
+  options: [30]

boxmot/configs/search/bytetrack.yaml

@@ -0,0 +1,13 @@
+# bytetrack_search_space.yaml
+
+track_thresh:
+  type: uniform
+  range: [0.4, 0.6]
+
+track_buffer:
+  type: randint
+  range: [10, 61, 10]  # The upper bound is exclusive, step size of 10
+
+match_thresh:
+  type: uniform
+  range: [0.7, 0.9]

boxmot/configs/search/deepocsort.yaml

@@ -0,0 +1,61 @@
+# deepocsort_search_space.yaml
+
+det_thresh:
+  type: uniform
+  range: [0.3, 0.6]
+
+max_age:
+  type: randint
+  range: [10, 61, 10]  # The upper bound is exclusive, step size of 10
+
+min_hits:
+  type: randint
+  range: [1, 6]  # The upper bound is exclusive
+
+iou_thresh:
+  type: uniform
+  range: [0.1, 0.4]
+
+delta_t:
+  type: randint
+  range: [1, 6]  # The upper bound is exclusive
+
+asso_func:
+  type: choice
+  options: ['iou', 'giou']
+
+inertia:
+  type: uniform
+  range: [0.1, 0.4]
+
+w_association_emb:
+  type: uniform
+  range: [0.5, 0.9]
+
+alpha_fixed_emb:
+  type: uniform
+  range: [0.9, 0.999]
+
+aw_param:
+  type: uniform
+  range: [0.3, 0.7]
+
+embedding_off:
+  type: choice
+  options: [True, False]
+
+cmc_off:
+  type: choice
+  options: [True, False]
+
+aw_off:
+  type: choice
+  options: [True, False]
+
+Q_xy_scaling:
+  type: uniform
+  range: [0.01, 1]
+
+Q_s_scaling:
+  type: uniform
+  range: [0.0001, 1]

boxmot/configs/search/hybridsort.yaml

@@ -0,0 +1,41 @@
+# hybridsort_search_space.yaml
+
+det_thresh:
+  type: uniform
+  range: [0, 0.6]
+
+max_age:
+  type: randint
+  range: [10, 151, 10]  # The upper bound is exclusive, step size of 10
+
+min_hits:
+  type: randint
+  range: [1, 6]  # The upper bound is exclusive
+
+delta_t:
+  type: randint
+  range: [1, 6]  # The upper bound is exclusive
+
+asso_func:
+  type: choice
+  options: ['iou', 'giou', 'diou']
+
+iou_thresh:
+  type: uniform
+  range: [0.1, 0.4]
+
+inertia:
+  type: uniform
+  range: [0.1, 0.4]
+
+TCM_first_step_weight:
+  type: uniform
+  range: [0, 0.5]
+
+longterm_reid_weight:
+  type: uniform
+  range: [0, 0.5]
+
+use_byte:
+  type: choice
+  options: [True, False]

boxmot/configs/search/imprassoc.yaml

@@ -0,0 +1,49 @@
+# imprassoc_search_space.yaml
+
+track_high_thresh:
+  type: uniform
+  range: [0.3, 0.7]
+
+track_low_thresh:
+  type: uniform
+  range: [0.1, 0.3]
+
+new_track_thresh:
+  type: uniform
+  range: [0.1, 0.8]
+
+track_buffer:
+  type: qrandint
+  range: [20, 80, 10]  # The upper bound is exclusive, step size of 10
+
+match_thresh:
+  type: uniform
+  range: [0.1, 0.9]
+
+second_match_thresh:
+  type: uniform
+  range: [0.1, 0.4]
+
+overlap_thresh:
+  type: uniform
+  range: [0.3, 0.6]
+
+proximity_thresh:
+  type: uniform
+  range: [0.1, 0.8]
+
+appearance_thresh:
+  type: uniform
+  range: [0.1, 0.8]
+
+cmc_method:
+  type: choice
+  options: ['sparseOptFlow']
+
+frame_rate:
+  type: choice
+  options: [30]
+
+lambda_:
+  type: uniform
+  range: [0.97, 0.995]

boxmot/configs/search/ocsort.yaml

@@ -0,0 +1,37 @@
+# ocsort_search_space.yaml
+
+det_thresh:
+  type: uniform
+  range: [0, 0.6]
+
+max_age:
+  type: grid_search
+  values: [10, 20, 30, 40, 50, 60]  # Discrete values using grid search
+
+min_hits:
+  type: grid_search
+  values: [1, 2, 3, 4, 5]  # Discrete values using grid search
+
+delta_t:
+  type: grid_search
+  values: [1, 2, 3, 4, 5]  # Discrete values using grid search
+
+asso_func:
+  type: choice
+  options: ['iou', 'giou', 'centroid']
+
+use_byte:
+  type: choice
+  options: [True, False]
+
+inertia:
+  type: uniform
+  range: [0.1, 0.4]
+
+Q_xy_scaling:
+  type: loguniform
+  range: [0.01, 1]
+
+Q_s_scaling:
+  type: loguniform
+  range: [0.0001, 1]

boxmot/configs/search/strongsort.yaml

@@ -0,0 +1,29 @@
+# strongsort_search_space.yaml
+
+ema_alpha:
+  type: uniform
+  range: [0.7, 0.95]
+
+max_cos_dist:
+  type: uniform
+  range: [0.1, 0.4]
+
+max_iou_dist:
+  type: uniform
+  range: [0.5, 0.95]
+
+max_age:
+  type: randint
+  range: [10, 151]  # The upper bound is exclusive
+
+n_init:
+  type: randint
+  range: [1, 4]  # The upper bound is exclusive
+
+mc_lambda:
+  type: uniform
+  range: [0.90, 0.999]
+
+nn_budget:
+  type: choice
+  options: [100]

boxmot/tracker_zoo.py

@@ -1,11 +1,11 @@
 # Mikel Broström 🔥 Yolo Tracking 🧾 AGPL-3.0 license
 
 import yaml
-from boxmot.utils import BOXMOT
+from boxmot.utils import BOXMOT, TRACKER_CONFIGS
 
 def get_tracker_config(tracker_type):
     """Returns the path to the tracker configuration file."""
-    return BOXMOT / 'configs' / f'{tracker_type}.yaml'
+    return TRACKER_CONFIGS / f'{tracker_type}.yaml'
 
 def create_tracker(tracker_type, tracker_config=None, reid_weights=None, device=None, half=None, per_class=None, evolve_param_dict=None):
     """

boxmot/utils/__init__.py

@@ -11,7 +11,8 @@
 DATA = ROOT / 'data'
 BOXMOT = ROOT / "boxmot"
 EXAMPLES = ROOT / "tracking"
-TRACKER_CONFIGS = ROOT / "boxmot" / "configs"
+TRACKER_CONFIGS = ROOT / "boxmot" / "configs" / "defaults"
+TRACKER_SEARCH_SPACES = ROOT / "boxmot" / "configs" / "search"
 WEIGHTS = ROOT / "tracking" / "weights"
 REQUIREMENTS = ROOT / "requirements.txt"