xqt commited on
Commit
00f618e
·
1 Parent(s): 52fdbff

FIX: added a forced post processing unit to Image Segmentation

Browse files
Files changed (1) hide show
  1. SegmentAnything2AssistApp.py +4 -0
SegmentAnything2AssistApp.py CHANGED
@@ -86,6 +86,10 @@ def __generate_mask(value, mask_threshold, max_hole_area, max_sprinkle_area, ima
86
  global __current_mask, __current_segment
87
  global __image_point_coords, __image_point_labels, __image_box
88
  global segment_anything2assist
 
 
 
 
89
  if VERBOSE:
90
  print("SegmentAnything2AssistApp::__generate_mask::Called.")
91
  mask_chw, mask_iou = segment_anything2assist.generate_masks_from_image(
 
86
  global __current_mask, __current_segment
87
  global __image_point_coords, __image_point_labels, __image_box
88
  global segment_anything2assist
89
+
90
+ # Force post processing of annotated image
91
+ __post_process_annotator_inputs(value)
92
+
93
  if VERBOSE:
94
  print("SegmentAnything2AssistApp::__generate_mask::Called.")
95
  mask_chw, mask_iou = segment_anything2assist.generate_masks_from_image(