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)