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
|