Merge pull request #1479 from YoungjaeDev/bug-evolve-deepocsort

Delete centroid from asso_func in deepocsort
mikel-brostrom · Jun 13, 2024 · b857c94 · b857c94
2 parents 9423fae + acfba6d
commit b857c94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tracking/evolve.py b/tracking/evolve.py
@@ -175,7 +175,7 @@ def get_new_config(self, trial):
             min_hits = trial.suggest_int("min_hits", 1, 5, step=1)
             iou_thresh = trial.suggest_float("iou_thresh", 0.1, 0.4)
             delta_t = trial.suggest_int("delta_t", 1, 5, step=1)
-            asso_func = trial.suggest_categorical("asso_func", ['iou', 'giou', 'centroid'])
+            asso_func = trial.suggest_categorical("asso_func", ['iou', 'giou'])
             inertia = trial.suggest_float("inertia", 0.1, 0.4)
             w_association_emb = trial.suggest_float("w_association_emb", 0.5, 0.9)
             alpha_fixed_emb = trial.suggest_float("alpha_fixed_emb", 0.9, 0.999)