Spaces:
Running
Running
jhj0517
commited on
Commit
·
381bc6f
1
Parent(s):
998201e
Add `invert_masks()`
Browse files- modules/mask_utils.py +5 -1
modules/mask_utils.py
CHANGED
@@ -17,6 +17,11 @@ def decode_to_mask(seg: np.ndarray[np.bool_] | np.ndarray[np.uint8]) -> np.ndarr
|
|
17 |
return seg.astype(np.uint8)
|
18 |
|
19 |
|
|
|
|
|
|
|
|
|
|
|
20 |
def generate_random_color() -> Tuple[int, int, int]:
|
21 |
"""Generate random color in RGB format"""
|
22 |
h = np.random.randint(0, 360)
|
@@ -47,7 +52,6 @@ def create_mask_layers(
|
|
47 |
List of RGBA images
|
48 |
"""
|
49 |
layer_list = []
|
50 |
-
|
51 |
sorted_masks = sorted(masks, key=lambda x: x['area'], reverse=True)
|
52 |
|
53 |
for info in sorted_masks:
|
|
|
17 |
return seg.astype(np.uint8)
|
18 |
|
19 |
|
20 |
+
def invert_masks(masks: List[Dict]) -> List[Dict]:
|
21 |
+
"""Invert the masks. Used for background masking"""
|
22 |
+
return [{'segmentation': 1 - mask['segmentation'], 'area': mask['area']} for mask in masks]
|
23 |
+
|
24 |
+
|
25 |
def generate_random_color() -> Tuple[int, int, int]:
|
26 |
"""Generate random color in RGB format"""
|
27 |
h = np.random.randint(0, 360)
|
|
|
52 |
List of RGBA images
|
53 |
"""
|
54 |
layer_list = []
|
|
|
55 |
sorted_masks = sorted(masks, key=lambda x: x['area'], reverse=True)
|
56 |
|
57 |
for info in sorted_masks:
|