Thiago Hersan commited on
Commit
0547194
1 Parent(s): f254011

maskformer-swin-large-coco

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -5,8 +5,10 @@ import numpy as np
5
  from transformers import MaskFormerFeatureExtractor, MaskFormerForInstanceSegmentation
6
 
7
 
8
- preprocessor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-tiny-ade")
9
- model = MaskFormerForInstanceSegmentation.from_pretrained("facebook/maskformer-swin-tiny-ade")
 
 
10
 
11
  def visualize_instance_seg_mask(mask):
12
  image = np.zeros((mask.shape[0], mask.shape[1], 3))
 
5
  from transformers import MaskFormerFeatureExtractor, MaskFormerForInstanceSegmentation
6
 
7
 
8
+ # preprocessor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-tiny-ade")
9
+ # model = MaskFormerForInstanceSegmentation.from_pretrained("facebook/maskformer-swin-tiny-ade")
10
+ preprocessor = MaskFormerFeatureExtractor.from_pretrained("facebook/maskformer-swin-large-coco")
11
+ model = MaskFormerForInstanceSegmentation.from_pretrained("facebook/maskformer-swin-large-coco")
12
 
13
  def visualize_instance_seg_mask(mask):
14
  image = np.zeros((mask.shape[0], mask.shape[1], 3))