changed the multimask_output to False to get only one mask from SAM

juglab · Nov 15, 2024 · 97d880d · 97d880d
1 parent 16c5e4e
commit 97d880d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/featureforest/postprocess/postprocess_with_sam.py b/src/featureforest/postprocess/postprocess_with_sam.py
@@ -139,7 +139,7 @@ def get_sam_mask(
             point_coords=None,
             point_labels=None,
             box=input_boxes[start:end],
-            multimask_output=True,
+            multimask_output=False,
         )
         reduced_axis = (0, 1) if masks.ndim == 4 else 0
         final_mask = np.bitwise_or(