From 4f48c78cfbcaded8c8b9957e8cd8eff5f7b68d7e Mon Sep 17 00:00:00 2001 From: crosstyan Date: Sun, 27 Apr 2025 17:00:56 +0800 Subject: [PATCH] fix: Update distance calculation in playground.py to include image size parameter - Modified the distance calculation function to accept an image size parameter, enhancing accuracy in 2D distance computations. - Improved the integration of camera parameters into the tracking logic for better performance and reliability. --- playground.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/playground.py b/playground.py index 779746e..5bfc764 100644 --- a/playground.py +++ b/playground.py @@ -762,7 +762,12 @@ for i, tracking in enumerate(trackings): tracking_2d_projection = camera.project(tracking.keypoints) for det in detections: delta_t = det.timestamp - tracking.last_active_timestamp - distance_2d = calculate_distance_2d(tracking_2d_projection, det.keypoints) + w, h = camera.params.image_size + distance_2d = calculate_distance_2d( + tracking_2d_projection, + det.keypoints, + image_size=(w, h), + ) affinity_2d = calculate_affinity_2d( distance_2d, delta_t,