Merge pull request #697 from roboflow/fix/sam2_producing_lines_as_seg…

…mentation_masks Fix issue with SAM2 producing segmentation masks as points and lines
roboflow · Sep 30, 2024 · dc9362b · dc9362b
2 parents 69bb0bc + 889ad6b
commit dc9362b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/inference/core/version.py b/inference/core/version.py
@@ -1,4 +1,4 @@
-__version__ = "0.21.0"
+__version__ = "0.21.1"
 
 
 if __name__ == "__main__":

diff --git a/inference/core/workflows/core_steps/models/foundation/segment_anything2/v1.py b/inference/core/workflows/core_steps/models/foundation/segment_anything2/v1.py
@@ -297,7 +297,7 @@ def convert_sam2_segmentation_response_to_inference_instances_seg_response(
         prompt_detection_ids,
     ):
         for mask in prediction.masks:
-            if len(mask) == 0:
+            if len(mask) < 3:
                 # skipping empty masks
                 continue
             if prediction.confidence < threshold: