Spaces:
Running
on
Zero
Running
on
Zero
FIX: added a forced post processing unit to Image Segmentation
Browse files
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(
|