search spaces now in yaml files

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/utils/__init__.py

@@ -12,6 +12,7 @@
 BOXMOT = ROOT / "boxmot"
 EXAMPLES = ROOT / "tracking"
 TRACKER_CONFIGS = ROOT / "boxmot" / "configs" / "defaults"
+TRACKER_SEARCH_SPACES = ROOT / "boxmot" / "configs" / "search"
 WEIGHTS = ROOT / "tracking" / "weights"
 REQUIREMENTS = ROOT / "requirements.txt"