Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ device = torch.device("cpu")
|
|
| 10 |
model = MaskFormerForInstanceSegmentation.from_pretrained("facebook/maskformer-swin-large-coco").to(device)
|
| 11 |
model.eval()
|
| 12 |
# preprocessor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-tiny-ade")
|
| 13 |
-
|
| 14 |
|
| 15 |
def visualize_instance_seg_mask(mask):
|
| 16 |
image = np.zeros((mask.shape[0], mask.shape[1], 3))
|
|
|
|
| 10 |
model = MaskFormerForInstanceSegmentation.from_pretrained("facebook/maskformer-swin-large-coco").to(device)
|
| 11 |
model.eval()
|
| 12 |
# preprocessor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-tiny-ade")
|
| 13 |
+
preprocessor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-large-coco")
|
| 14 |
|
| 15 |
def visualize_instance_seg_mask(mask):
|
| 16 |
image = np.zeros((mask.shape[0], mask.shape[1], 3))
|