Jingkang Yang commited on
Commit
46a52de
1 Parent(s): aaf4c51

update: app

Browse files
Files changed (1) hide show
  1. open_vocab_seg/utils/predictor.py +2 -2
open_vocab_seg/utils/predictor.py CHANGED
@@ -179,7 +179,7 @@ class VisualizationDemo(object):
179
  if "sem_seg" in predictions:
180
  r = predictions["sem_seg"]
181
  pred_mask = r.argmax(dim=0).to('cpu')
182
- pred_mask = np.array(pred_mask, dtype=np.int)
183
 
184
  vis_output = visualizer.draw_sem_seg(
185
  pred_mask
@@ -239,7 +239,7 @@ class VisualizationDemo(object):
239
  if "sem_seg" in predictions:
240
  r = predictions["sem_seg"]
241
  pred_mask = r.argmax(dim=0).to('cpu')
242
- pred_mask = np.array(pred_mask, dtype=np.int)
243
 
244
  pred_mask_sam_rgb = pred_mask.copy()
245
  for mask in masks_rgb:
 
179
  if "sem_seg" in predictions:
180
  r = predictions["sem_seg"]
181
  pred_mask = r.argmax(dim=0).to('cpu')
182
+ pred_mask = np.array(pred_mask, dtype=int)
183
 
184
  vis_output = visualizer.draw_sem_seg(
185
  pred_mask
 
239
  if "sem_seg" in predictions:
240
  r = predictions["sem_seg"]
241
  pred_mask = r.argmax(dim=0).to('cpu')
242
+ pred_mask = np.array(pred_mask, dtype=int)
243
 
244
  pred_mask_sam_rgb = pred_mask.copy()
245
  for mask in masks_rgb: