Update mask_adapter/sam_maskadapter.py

mask_adapter/sam_maskadapter.py

@@ -139,7 +139,8 @@ class SAMVisualizationDemo(object):
         image = (image - pixel_mean) / pixel_std
 
         image = image.unsqueeze(0)
-        
+
+        image = image.to(text_features)
         # if len(class_names) == 1:
         #     class_names.append('others')
         # txts = [f'a photo of {cls_name}' for cls_name in class_names]
@@ -305,7 +306,7 @@ class SAMPointVisualizationDemo(object):
 
         # txts = [f'a photo of {cls_name}' for cls_name in self.class_names]
         # text = open_clip.tokenize(txts)
-
+        
         with torch.no_grad():
             # text_features = self.clip_model.encode_text(text.cuda())
             # text_features /= text_features.norm(dim=-1, keepdim=True)