Spaces:

martintomov
/

InsectSAM

Running on Zero

Martin Tomov commited on May 16, 2024

Commit

2131407

verified ·

1 Parent(s): a9448c9

keycv2 fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -91,7 +91,7 @@ def mask_to_polygon(mask: np.ndarray) -> np.ndarray:
     contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
     if len(contours) == 0:
         return np.array([])
-    largest_contour = max(contours, keycv2.contourArea)
     return largest_contour
 def refine_masks(masks: torch.BoolTensor, polygon_refinement: bool = False) -> List[np.ndarray]:

     contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
     if len(contours) == 0:
         return np.array([])
+    largest_contour = max(contours, key=cv2.contourArea)
     return largest_contour
 def refine_masks(masks: torch.BoolTensor, polygon_refinement: bool = False) -> List[np.ndarray]: