okooooooo/supervision
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

check for empty mask

Browse Source
This commit is contained in:
magda skoczen committed 2024-05-16 09:23:22 +02:00
2 parents 3ee72e38c3 ce05a0c544
commit 511f0ab9b2
1 file changed
+2 -3
Download Patch File Download Diff File Expand all files Collapse all files
supervision/dataset/formats/coco.py
+2 -3
View File
@@ -108,9 +108,8 @@ def coco_annotations_to_detections(
def _mask_has_holes(mask: np.ndarray) -> bool:
mask_uint8 = mask.astype(np.uint8)
_, hierarchy = cv2.findContours(
mask_uint8, cv2.RETR_CCOMP, cv2.CHAIN_APPROX_SIMPLE)
mask_uint8 = mask.astype(np.uint8)
_, hierarchy = cv2.findContours(mask_uint8, cv2.RETR_CCOMP, cv2.CHAIN_APPROX_SIMPLE)
parent_countour_index = 3
for h in hierarchy[0]:
if h[parent_countour_index] != -1: