fix: add timestamps to outgoing messages

luxonis · Aug 16, 2024 · 5d7da5b · 5d7da5b
1 parent ec9ad5a
commit 5d7da5b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/depthai_nodes/ml/parsers/age_gender.py b/depthai_nodes/ml/parsers/age_gender.py
@@ -38,5 +38,6 @@ def run(self):
             prob = output.getTensor("prob", dequantize=True).flatten().tolist()
 
             age_gender_message = create_age_gender_message(age=age, gender_prob=prob)
+            age_gender_message.setTimestamp(output.getTimestamp())
 
             self.out.send(age_gender_message)
diff --git a/depthai_nodes/ml/parsers/hrnet.py b/depthai_nodes/ml/parsers/hrnet.py
@@ -70,10 +70,11 @@ def run(self):
                 [map_w, map_h]
             )  # normalize keypoints to [0, 1]
 
-            keypoints_msg = create_keypoints_message(
+            keypoints_message = create_keypoints_message(
                 keypoints=keypoints,
                 scores=scores,
                 confidence_threshold=self.score_threshold,
             )
+            keypoints_message.setTimestamp(output.getTimestamp())
 
-            self.out.send(keypoints_msg)
+            self.out.send(keypoints_message)