Spaces:

facebook
/

ov-seg

Runtime error

JeffLiang commited on Apr 10, 2023

Commit

01d0cdf

•

1 Parent(s): 9d3c696

bug fix

Files changed (1) hide show

open_vocab_seg/modeling/clip_adapter/utils.py CHANGED Viewed

@@ -62,7 +62,7 @@ def crop_with_mask(
     new_image = torch.cat(
         [image.new_full((1, b - t, r - l), fill_value=val) for val in fill]
     )
-    # return image[:, t:b, l:r], mask[None, t:b, l:r]
     return image[:, t:b, l:r] * mask[None, t:b, l:r] + (~ mask[None, t:b, l:r]) * new_image, mask[None, t:b, l:r]

     new_image = torch.cat(
         [image.new_full((1, b - t, r - l), fill_value=val) for val in fill]
     )
+    mask = mask.bool()
     return image[:, t:b, l:r] * mask[None, t:b, l:r] + (~ mask[None, t:b, l:r]) * new_image, mask[None, t:b, l:r]