Hugging Face's logo

Spaces:
wondervictor
/
Mask-Adapter
Running on Zero

App Files Community
wondervictor commited on
Commit
e8f0644
·
verified ·
1 Parent(s): bcf38fb

Update mask_adapter/modeling/meta_arch/mask_adapter_head.py

Browse files
Files changed (1) hide show
  1. mask_adapter/modeling/meta_arch/mask_adapter_head.py +1 -1
mask_adapter/modeling/meta_arch/mask_adapter_head.py CHANGED
@@ -86,7 +86,7 @@ class MASKAdapterHead(nn.Module):
86
  clip_feature = repeat(clip_feature, "B C H W -> (B N) C H W", N=N)
87
 
88
  H,W = clip_feature.shape[-2:]
89
- masks = F.interpolate(masks.float(), size=(H*4,W*4),
90
  mode='bilinear', align_corners=False)
91
  masks = self.mask_downscaling(masks)
92
 
 
86
  clip_feature = repeat(clip_feature, "B C H W -> (B N) C H W", N=N)
87
 
88
  H,W = clip_feature.shape[-2:]
89
+ masks = F.interpolate(masks, size=(H*4,W*4),
90
  mode='bilinear', align_corners=False)
91
  masks = self.mask_downscaling(masks)
92