jhj0517 commited on
Commit
381bc6f
·
1 Parent(s): 998201e

Add `invert_masks()`

Browse files
Files changed (1) hide show
  1. 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: