Update mask_adapter/sam_maskadapter.py

mask_adapter/sam_maskadapter.py

@@ -188,7 +188,7 @@ class SAMVisualizationDemo(object):
                 select_mask.extend(locs[0].tolist())
         for idx in select_mask:
             select_cls[idx] = class_preds[idx]
-        semseg = torch.einsum("qc,qhw->chw", select_cls.float(), pred_masks.tensor.float().cuda())
+        semseg = torch.einsum("qc,qhw->chw", select_cls.float(), pred_masks.tensor.float().cpu())
 
         r = semseg
         blank_area = (r[0] == 0)
@@ -239,7 +239,7 @@ class SAMPointVisualizationDemo(object):
         lvis_classes = [x[x.find(':')+1:] for x in lvis_classes]
                 
         self.class_names = thing_classes + stuff_classes + lvis_classes
-        self.text_embedding = torch.from_numpy(np.load("./text_embedding/lvis_coco_text_embedding.npy")).to("cuda")
+        self.text_embedding = torch.from_numpy(np.load("./text_embedding/lvis_coco_text_embedding.npy")).to("cpu")
     
         self.class_names = self._load_class_names()