From ded4392a24ecda9d1699f11c4a45803f7355ca14 Mon Sep 17 00:00:00 2001
From: kkeroo <61207502+kkeroo@users.noreply.github.com>
Date: Thu, 29 Aug 2024 11:53:30 +0200
Subject: [PATCH] Normalize detections.

---
 depthai_nodes/ml/parsers/utils/scrfd.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/depthai_nodes/ml/parsers/utils/scrfd.py b/depthai_nodes/ml/parsers/utils/scrfd.py
index 533f4a05..6e8586e2 100644
--- 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