Better error in MPPalmDetectionParser. (#72)

luxonis · Sep 19, 2024 · 59a0190 · 59a0190
1 parent 7345faa
commit 59a0190
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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