Spaces:
Runtime error
Runtime error
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)
|