Add normalization.

luxonis · Sep 3, 2024 · a15c9f6 · a15c9f6
1 parent b7b0d5c
commit a15c9f6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/depthai_nodes/ml/parsers/mediapipe_palm_detection.py b/depthai_nodes/ml/parsers/mediapipe_palm_detection.py
@@ -148,6 +148,8 @@ def run(self):
             bboxes = np.array(bboxes)[indices]
             scores = np.array(scores)[indices]
 
+            bboxes = bboxes.astype(np.float32) / self.scale
+
             detections_msg = create_detection_message(bboxes, scores, labels=None)
             detections_msg.setTimestamp(output.getTimestamp())
             self.out.send(detections_msg)