From f62ce478230ea32fb09a85ea7bf953c527dbc57a Mon Sep 17 00:00:00 2001 From: Miguelmelon Date: Tue, 28 May 2024 14:43:05 +0200 Subject: [PATCH] Deleting previous HoC detections --- people_tracking_v2/scripts/comparison_node.py | 7 +++++-- people_tracking_v2/scripts/matching_node.py | 0 2 files changed, 5 insertions(+), 2 deletions(-) create mode 100644 people_tracking_v2/scripts/matching_node.py diff --git a/people_tracking_v2/scripts/comparison_node.py b/people_tracking_v2/scripts/comparison_node.py index 082a964..271f9c6 100755 --- a/people_tracking_v2/scripts/comparison_node.py +++ b/people_tracking_v2/scripts/comparison_node.py @@ -74,12 +74,12 @@ def compare_data(self): if not self.latest_hoc_vectors or self.latest_pose_data is None or self.saved_pose_data is None: return - for hoc_vector in self.latest_hoc_vectors: + for i, hoc_vector in enumerate(self.latest_hoc_vectors): # Compare HoC data hue_vector = hoc_vector.hue_vector sat_vector = hoc_vector.sat_vector hoc_distance_score = self.compute_hoc_distance_score(hue_vector, sat_vector) - rospy.loginfo(f"HoC Distance score for detection: {hoc_distance_score}") + rospy.loginfo(f"Detection #{i+1}: HoC Distance score: {hoc_distance_score}") # Compare pose data left_shoulder_hip_distance = self.latest_pose_data.left_shoulder_hip_distance @@ -94,6 +94,9 @@ def compare_data(self): # Publish debug information self.publish_debug_info(hoc_distance_score, pose_distance_score) + + # Clear the latest HoC vectors after processing + self.latest_hoc_vectors.clear() def compute_hoc_distance_score(self, hue_vector, sat_vector): """Compute the distance score between the current detection and saved data (HoC).""" diff --git a/people_tracking_v2/scripts/matching_node.py b/people_tracking_v2/scripts/matching_node.py new file mode 100644 index 0000000..e69de29