radames commited on
Commit
e0ea5c4
1 Parent(s): 9aadb9a

remove check

Browse files
Files changed (1) hide show
  1. model.py +0 -2
model.py CHANGED
@@ -90,8 +90,6 @@ class Model:
90
  seg_map = np.full(mask.shape[:-1], -1)
91
  for index, color in enumerate(COLOR_LIST):
92
  seg_map[np.sum(mask == color, axis=2) == 3] = index
93
- if not (seg_map != -1).all():
94
- return None
95
  return seg_map
96
 
97
  @staticmethod
 
90
  seg_map = np.full(mask.shape[:-1], -1)
91
  for index, color in enumerate(COLOR_LIST):
92
  seg_map[np.sum(mask == color, axis=2) == 3] = index
 
 
93
  return seg_map
94
 
95
  @staticmethod