Gaurav commited on
Commit
2d796b3
1 Parent(s): 4b53639

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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}