Normalize detections. (#31)

luxonis · Aug 30, 2024 · 07923a1 · 07923a1
1 parent 9b4aff5
commit 07923a1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/depthai_nodes/ml/parsers/utils/scrfd.py b/depthai_nodes/ml/parsers/utils/scrfd.py
@@ -166,9 +166,11 @@ def decode_scrfd(
     kpss = kpss[order, :, :]
     kpss = kpss[keep, :, :]
 
+    height, width = input_size
     scores = det[:, 4]
-    bboxes = np.int32(det[:, :4])
-    keypoints = np.int32(kpss)
+    bboxes = det[:, :4] / np.array([width, height] * 2)
+
+    keypoints = kpss / np.tile([width, height], (5, 1))
     keypoints = keypoints.reshape(-1, 5, 2)
 
     return bboxes, scores, keypoints