Spaces:
Running
on
Zero
Running
on
Zero
Update mask_adapter/sam_maskadapter.py
Browse files
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)
|