Enable min object size of up to 10000 in auto segmentation widget

computational-cell-analytics · Sep 2, 2023 · 9bad755 · 9bad755
1 parent 1d11930
commit 9bad755
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/micro_sam/sam_annotator/annotator_2d.py b/micro_sam/sam_annotator/annotator_2d.py
@@ -47,7 +47,10 @@ def _changed_param(amg, **params):
     return None
 
 
-@magicgui(call_button="Automatic Segmentation")
+@magicgui(
+    call_button="Automatic Segmentation",
+    min_object_size={"min": 0, "max": 10000},
+)
 def _autosegment_widget(
     v: Viewer,
     pred_iou_thresh: float = 0.88,

diff --git a/micro_sam/sam_annotator/annotator_3d.py b/micro_sam/sam_annotator/annotator_3d.py
@@ -134,7 +134,10 @@ def _segment_volume_widget(
     v.layers[layer].refresh()
 
 
-@magicgui(call_button="Automatic Segmentation")
+@magicgui(
+    call_button="Automatic Segmentation",
+    min_object_size={"min": 0, "max": 10000}
+)
 def _autosegment_widget(
     v: Viewer,
     pred_iou_thresh: float = 0.88,