update: predictor

open_vocab_seg/utils/predictor.py

@@ -178,7 +178,7 @@ class VisualizationDemo(object):
         if "sem_seg" in predictions:
             r = predictions["sem_seg"]
             pred_mask = r.argmax(dim=0).to('cpu')
-            pred_mask = np.array(pred_mask, dtype=np.int)
+            pred_mask = np.array(pred_mask, dtype=int)
 
             vis_output = visualizer.draw_sem_seg(
                 pred_mask
@@ -236,7 +236,7 @@ class VisualizationDemo(object):
         if "sem_seg" in predictions:
             r = predictions["sem_seg"]
             pred_mask = r.argmax(dim=0).to('cpu')
-            pred_mask = np.array(pred_mask, dtype=np.int)
+            pred_mask = np.array(pred_mask, dtype=int)
             
             pred_mask_sam_rgb = pred_mask.copy()
             for mask in masks_rgb: