chore: change distance function thresholds

demo_utils/configuration.py

@@ -1,4 +1,5 @@
-DISTANCE_THRESHOLD_CENTROID: int = 0.08
+DISTANCE_THRESHOLD_CENTROID: int = 0.8
+DISTANCE_THRESHOLD_BBOX: int = 0.7
 MAX_DISTANCE: int = 10000
 
 models_path = {"YOLOv7": "custom_models/yolov7.pt", "YOLOv7 Tiny": "custom_models/yolov7-tiny.pt"}
@@ -6,7 +7,7 @@ models_path = {"YOLOv7": "custom_models/yolov7.pt", "YOLOv7 Tiny": "custom_model
 style = {"Bounding box": "bbox", "Centroid": "centroid"}
 
 examples = {
-    "soccer": {"distance_threshold": 0.1, "absolute_path": True, "classes": [0]},
-    "oxford_town_center": {"distance_threshold": 0.05, "absolute_path": False, "classes": [0]},
-    "traffic_1_A": {"distance_threshold": 0.1, "absolute_path": False, "classes": [2, 3, 5, 7]},
+    "soccer": {"distance_threshold": 0.7, "absolute_path": True, "classes": [0]},
+    "oxford_town_center": {"distance_threshold": 0.7, "absolute_path": False, "classes": [0]},
+    "traffic_1_A": {"distance_threshold": 0.7, "absolute_path": False, "classes": [2, 3, 5, 7]},
 }

inference.py

@@ -5,6 +5,7 @@ from norfair.distances import create_normalized_mean_euclidean_distance
 
 from custom_models import YOLO, yolo_detections_to_norfair_detections
 from demo_utils.configuration import (
+    DISTANCE_THRESHOLD_BBOX,
     DISTANCE_THRESHOLD_CENTROID,
     examples,
     models_path,
@@ -44,7 +45,11 @@ def inference(
         )
 
     distance_function = "iou" if track_points == style["Bounding box"] else "euclidean"
-    distance_threshold = DISTANCE_THRESHOLD_CENTROID
+    distance_threshold = (
+        DISTANCE_THRESHOLD_BBOX
+        if track_points == style["Bounding box"]
+        else DISTANCE_THRESHOLD_CENTROID
+    )
 
     if motion_estimation and drawing_paths:
         fix_paths = True