Update mask_adapter/sam_maskadapter.py

mask_adapter/sam_maskadapter.py

@@ -150,7 +150,7 @@ class SAMVisualizationDemo(object):
             # text_features = self.clip_model.encode_text(text)
             # text_features /= text_features.norm(dim=-1, keepdim=True)
             
-            features = self.extract_features_convnext(image)
+            features = self.extract_features_convnext(image.half())
             
             clip_feature = features['clip_vis_dense']
             
@@ -311,7 +311,7 @@ class SAMPointVisualizationDemo(object):
             # text_features /= text_features.norm(dim=-1, keepdim=True)
             #np.save("/home/yongkangli/Mask-Adapter/text_embedding/lvis_coco_text_embedding.npy", text_features.cpu().numpy())
             text_features = self.text_embedding
-            features = self.extract_features_convnext(image)
+            features = self.extract_features_convnext(image.half())
             clip_feature = features['clip_vis_dense']
 
             clip_vis_dense = self.visual_prediction_forward_convnext_2d(clip_feature)