From ef9453993dd8d2670b314170e1b4c56d1246be6b Mon Sep 17 00:00:00 2001 From: kkeroo <61207502+kkeroo@users.noreply.github.com> Date: Fri, 23 Aug 2024 15:10:06 +0200 Subject: [PATCH] Better bbox calculation. --- depthai_nodes/ml/parsers/mediapipe_palm_detection.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/depthai_nodes/ml/parsers/mediapipe_palm_detection.py b/depthai_nodes/ml/parsers/mediapipe_palm_detection.py index d00e92d..e3d21eb 100644 --- a/depthai_nodes/ml/parsers/mediapipe_palm_detection.py +++ b/depthai_nodes/ml/parsers/mediapipe_palm_detection.py @@ -105,10 +105,10 @@ def run(self): for hand in decoded_bboxes: extended_points = hand.rect_points - xmin = int(min(extended_points[0][0], extended_points[1][0])) - ymin = int(min(extended_points[0][1], extended_points[1][1])) - xmax = int(max(extended_points[2][0], extended_points[3][0])) - ymax = int(max(extended_points[2][1], extended_points[3][1])) + xmin = int(min([point[0] for point in extended_points])) + ymin = int(min([point[1] for point in extended_points])) + xmax = int(max([point[0] for point in extended_points])) + ymax = int(max([point[1] for point in extended_points])) bboxes.append([xmin, ymin, xmax, ymax]) scores.append(hand.pd_score)