From 735d85a349bebb0ec31d993fa863458a8640d38d Mon Sep 17 00:00:00 2001 From: jkbmrz Date: Fri, 16 Aug 2024 14:49:58 +0200 Subject: [PATCH] fix: add timestamps to outgoing messages --- depthai_nodes/ml/parsers/age_gender.py | 1 + depthai_nodes/ml/parsers/hrnet.py | 5 +++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/depthai_nodes/ml/parsers/age_gender.py b/depthai_nodes/ml/parsers/age_gender.py index e2fd7cd4..199f840e 100644 --- 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 index 7d1b83b0..be91d1c9 100644 --- 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)