Merge pull request #1668 from mikel-brostrom/merge-default-configs-n-…

…search-spaces

merge configs and search spaces in single yamls

boxmot/configs/search/botsort.yaml → boxmot/configs/botsort.yaml

@@ -1,37 +1,34 @@
-# botsort_search_space.yaml
-
 track_high_thresh:
   type: uniform
+  default: 0.6  # from the default parameters
   range: [0.3, 0.7]
 
 track_low_thresh:
   type: uniform
+  default: 0.1  # from the default parameters
   range: [0.1, 0.3]
 
 new_track_thresh:
   type: uniform
+  default: 0.7  # from the default parameters
   range: [0.1, 0.8]
 
 track_buffer:
   type: randint
+  default: 30  # from the default parameters
   range: [20, 81]
 
 match_thresh:
   type: uniform
+  default: 0.8  # from the default parameters
   range: [0.1, 0.9]
 
 proximity_thresh:
   type: uniform
+  default: 0.5  # from the default parameters
   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]
+  default: 0.25  # from the default parameters
+  range: [0.1, 0.8]

boxmot/configs/bytetrack.yaml

@@ -0,0 +1,19 @@
+track_thresh:
+  type: uniform
+  default: 0.5  # from the default parameters
+  range: [0.4, 0.6]
+
+track_buffer:
+  type: randint
+  default: 30  # from the default parameters
+  range: [10, 61, 10]  # step size of 10, upper bound exclusive
+
+match_thresh:
+  type: uniform
+  default: 0.8  # from the default parameters
+  range: [0.7, 0.9]
+
+frame_rate:
+  type: choice
+  default: 30  # from the default parameters
+  choices: [30]  # static choice for Ray Search

boxmot/configs/deepocsort.yaml

@@ -0,0 +1,74 @@
+det_thresh:
+  type: uniform
+  default: 0.5  # from the default parameters
+  range: [0.3, 0.6]
+
+max_age:
+  type: randint
+  default: 30  # from the default parameters
+  range: [10, 61, 10]  # step size of 10, upper bound exclusive
+
+min_hits:
+  type: randint
+  default: 3  # from the default parameters
+  range: [1, 6]  # upper bound exclusive
+
+iou_thresh:
+  type: uniform
+  default: 0.3  # from the default parameters
+  range: [0.1, 0.4]
+
+delta_t:
+  type: randint
+  default: 3  # from the default parameters
+  range: [1, 6]  # upper bound exclusive
+
+asso_func:
+  type: choice
+  default: iou  # from the default parameters
+  options: ['iou', 'giou']
+
+inertia:
+  type: uniform
+  default: 0.2  # from the default parameters
+  range: [0.1, 0.4]
+
+w_association_emb:
+  type: uniform
+  default: 0.75  # from the default parameters
+  range: [0.5, 0.9]
+
+alpha_fixed_emb:
+  type: uniform
+  default: 0.95  # from the default parameters
+  range: [0.9, 0.999]
+
+aw_param:
+  type: uniform
+  default: 0.5  # from the default parameters
+  range: [0.3, 0.7]
+
+embedding_off:
+  type: choice
+  default: false  # from the default parameters
+  options: [True, False]
+
+cmc_off:
+  type: choice
+  default: false  # from the default parameters
+  options: [True, False]
+
+aw_off:
+  type: choice
+  default: false  # from the default parameters
+  options: [True, False]
+
+Q_xy_scaling:
+  type: uniform
+  default: 0.01  # from the default parameters
+  range: [0.01, 1]
+
+Q_s_scaling:
+  type: uniform
+  default: 0.0001  # from the default parameters
+  range: [0.0001, 1]

boxmot/configs/hybridsort.yaml

@@ -0,0 +1,49 @@
+det_thresh:
+  type: uniform
+  default: 0.12442660055370669  # from the default parameters
+  range: [0, 0.6]
+
+max_age:
+  type: randint
+  default: 30  # from the default parameters
+  range: [10, 151, 10]  # step size of 10, upper bound exclusive
+
+min_hits:
+  type: randint
+  default: 1  # from the default parameters
+  range: [1, 6]  # upper bound exclusive
+
+delta_t:
+  type: randint
+  default: 5  # from the default parameters
+  range: [1, 6]  # upper bound exclusive
+
+asso_func:
+  type: choice
+  default: hmiou  # from the default parameters
+  options: ['iou', 'giou', 'diou']
+
+iou_threshold:
+  type: uniform
+  default: 0.3  # from the default parameters
+  range: [0.1, 0.4]
+
+inertia:
+  type: uniform
+  default: 0.369525477649008  # from the default parameters
+  range: [0.1, 0.4]
+
+TCM_first_step_weight:
+  type: uniform
+  default: 0.2866529225304586  # from the default parameters
+  range: [0, 0.5]
+
+longterm_reid_weight:
+  type: uniform
+  default: 0.0509704360503877  # from the default parameters
+  range: [0, 0.5]
+
+use_byte:
+  type: choice
+  default: False  # from the default parameters
+  options: [True, False]

boxmot/configs/imprassoc.yaml

@@ -0,0 +1,59 @@
+track_high_thresh:
+  type: uniform
+  default: 0.5  # from the default parameters
+  range: [0.3, 0.7]
+
+track_low_thresh:
+  type: uniform
+  default: 0.1  # from the default parameters
+  range: [0.05, 0.3]
+
+new_track_thresh:
+  type: uniform
+  default: 0.5  # from the default parameters
+  range: [0.5, 0.9]
+
+track_buffer:
+  type: qrandint
+  default: 35  # from the default parameters
+  range: [20, 80, 10]  # step size of 10, upper bound exclusive
+
+match_thresh:
+  type: uniform
+  default: 0.65  # from the default parameters
+  range: [0.1, 0.9]
+
+second_match_thresh:
+  type: uniform
+  default: 0.19  # from the default parameters
+  range: [0.1, 0.4]
+
+overlap_thresh:
+  type: uniform
+  default: 0.55  # from the default parameters
+  range: [0.3, 0.6]
+
+proximity_thresh:
+  type: uniform
+  default: 0.1  # from the default parameters
+  range: [0.1, 0.8]
+
+appearance_thresh:
+  type: uniform
+  default: 0.25  # from the default parameters
+  range: [0.1, 0.8]
+
+cmc_method:
+  type: choice
+  default: sparseOptFlow  # from the default parameters
+  options: ['sparseOptFlow']
+
+frame_rate:
+  type: choice
+  default: 30  # from the default parameters
+  options: [30]
+
+lambda_:
+  type: uniform
+  default: 0.05  # from the default parameters
+  range: [0.05, 0.3]

boxmot/configs/ocsort.yaml

@@ -0,0 +1,44 @@
+det_thresh:
+  type: uniform
+  default: 0.6  # from the default parameters
+  range: [0, 0.6]
+
+max_age:
+  type: grid_search
+  default: 30  # from the default parameters
+  values: [10, 20, 30, 40, 50, 60]
+
+min_hits:
+  type: grid_search
+  default: 3  # from the default parameters
+  values: [1, 2, 3, 4, 5]
+
+delta_t:
+  type: grid_search
+  default: 3  # from the default parameters
+  values: [1, 2, 3, 4, 5]
+
+asso_func:
+  type: choice
+  default: iou  # from the default parameters
+  options: ['iou', 'giou', 'centroid']
+
+use_byte:
+  type: choice
+  default: false  # from the default parameters
+  options: [True, False]
+
+inertia:
+  type: uniform
+  default: 0.2  # from the default parameters
+  range: [0.1, 0.4]
+
+Q_xy_scaling:
+  type: loguniform
+  default: 0.01  # from the default parameters
+  range: [0.01, 1]
+
+Q_s_scaling:
+  type: loguniform
+  default: 0.0001  # from the default parameters
+  range: [0.0001, 1]