Spaces:
Runtime error
Runtime error
remove check
Browse files
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
|