Spaces:
Runtime error
Runtime error
Gaurav
commited on
Commit
•
2d796b3
1
Parent(s):
4b53639
Update app.py
Browse files
app.py
CHANGED
@@ -10,6 +10,7 @@ model.eval()
|
|
10 |
preprocessor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-tiny-ade")
|
11 |
|
12 |
def visualize_instance_seg_mask(mask):
|
|
|
13 |
image = np.zeros((mask.shape[0], mask.shape[1], 3))
|
14 |
labels = np.unique(mask)
|
15 |
label2color = {label: (random.randint(0, 1), random.randint(0, 255), random.randint(0, 255)) for label in labels}
|
|
|
10 |
preprocessor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-tiny-ade")
|
11 |
|
12 |
def visualize_instance_seg_mask(mask):
|
13 |
+
print(mask)
|
14 |
image = np.zeros((mask.shape[0], mask.shape[1], 3))
|
15 |
labels = np.unique(mask)
|
16 |
label2color = {label: (random.randint(0, 1), random.randint(0, 255), random.randint(0, 255)) for label in labels}
|