From 59a0190c75cc2c8c58e66b75552a2e649579cd2c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ja=C5=A1a=20Kerec?= <61207502+kkeroo@users.noreply.github.com> Date: Thu, 19 Sep 2024 10:07:57 +0200 Subject: [PATCH] Better error in MPPalmDetectionParser. (#72) --- depthai_nodes/ml/parsers/utils/medipipe.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/depthai_nodes/ml/parsers/utils/medipipe.py b/depthai_nodes/ml/parsers/utils/medipipe.py index 8d0234b..69c9845 100644 --- a/depthai_nodes/ml/parsers/utils/medipipe.py +++ b/depthai_nodes/ml/parsers/utils/medipipe.py @@ -240,8 +240,13 @@ def decode_bboxes(score_thresh, scores, bboxes, anchors, scale=128, best_only=Fa det_scores = scores[detection_mask] if det_scores.size == 0: return regions + try: det_bboxes2 = bboxes[detection_mask] det_anchors = anchors[detection_mask] + except Exception as e: + raise IndexError( + "Wrong parser scale set. Please use setScale method to set different scale according to model dimensions (e.g. 128)." + ) from e det_bboxes = det_bboxes2 * np.tile(det_anchors[:, 2:4], 9) / scale + np.tile( det_anchors[:, 0:2], 9