Resolve some of corner-cases

sovrasov · sovrasov · commit 91e5fff145b2 · 2025-07-25T21:25:22.000+09:00
diff --git a/src/python/model_api/models/segmentation.py b/src/python/model_api/models/segmentation.py
@@ -212,7 +212,8 @@ def get_contours(
                 cv2.RETR_CCOMP,
                 cv2.CHAIN_APPROX_NONE,
             )
-            hierarchy = hierarchy.squeeze()
+            if len(contours):
+                hierarchy = hierarchy.squeeze(axis=0)
 
             for i, contour in enumerate(contours):
                 if hierarchy[i][3] >= 0:

Original file line number	Diff line number	Diff line change
`@@ -212,7 +212,8 @@ def get_contours(`
`212`	`212`	`cv2.RETR_CCOMP,`
`213`	`213`	`cv2.CHAIN_APPROX_NONE,`
`214`	`214`	`)`
`215`		`- hierarchy = hierarchy.squeeze()`
	`215`	`+ if len(contours):`
	`216`	`+ hierarchy = hierarchy.squeeze(axis=0)`
`216`	`217`
`217`	`218`	`for i, contour in enumerate(contours):`
`218`	`219`	`if hierarchy[i][3] >= 0:`