Update distance functions

inference.py

@@ -43,9 +43,7 @@ def inference(
             max_points=500, min_distance=7, transformations_getter=transformations_getter
         )
 
-    distance_function = create_normalized_mean_euclidean_distance(
-        video.input_height, video.input_width
-    )
+    distance_function = "iou" if track_points == style["Bounding box"] else "euclidean"
     distance_threshold = DISTANCE_THRESHOLD_CENTROID
 
     if motion_estimation and drawing_paths:
@@ -90,13 +88,12 @@ def inference(
             detections=detections, coord_transformations=coord_transformations
         )
 
-        mask = np.ones(frame.shape[:2], frame.dtype)
-        if track_points == "bbox":
-            for det in detections:
-                i = det.points.astype(int)
-                mask[i[0, 1] : i[1, 1], i[0, 0] : i[1, 0]] = 0
-
         if motion_estimation:
+            mask = np.ones(frame.shape[:2], frame.dtype)
+            if track_points == "bbox":
+                for det in detections:
+                    i = det.points.astype(int)
+                    mask[i[0, 1] : i[1, 1], i[0, 0] : i[1, 0]] = 0
             coord_transformations = motion_estimator.update(frame, mask)
 
         frame = draw(