wondervictor commited on
Commit
ccba03f
·
verified ·
1 Parent(s): e8b794f

Update mask_adapter/sam_maskadapter.py

Browse files
Files changed (1) hide show
  1. mask_adapter/sam_maskadapter.py +1 -1
mask_adapter/sam_maskadapter.py CHANGED
@@ -316,7 +316,7 @@ class SAMPointVisualizationDemo(object):
316
 
317
  clip_vis_dense = self.visual_prediction_forward_convnext_2d(clip_feature)
318
 
319
- semantic_activation_maps = self.mask_adapter(clip_vis_dense, pred_masks.tensor.unsqueeze(0).float()))
320
  maps_for_pooling = F.interpolate(semantic_activation_maps, size=clip_feature.shape[-2:], mode='bilinear', align_corners=False)
321
 
322
  B, C = clip_feature.size(0), clip_feature.size(1)
 
316
 
317
  clip_vis_dense = self.visual_prediction_forward_convnext_2d(clip_feature)
318
 
319
+ semantic_activation_maps = self.mask_adapter(clip_vis_dense, pred_masks.tensor.unsqueeze(0).float())
320
  maps_for_pooling = F.interpolate(semantic_activation_maps, size=clip_feature.shape[-2:], mode='bilinear', align_corners=False)
321
 
322
  B, C = clip_feature.size(0), clip_feature.size(1)