Spaces:
Build error
Build error
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(
|