Spaces:

facebook
/

ov-seg

Runtime error

JeffLiang commited on Apr 10, 2023

Commit

9d3c696

1 Parent(s): 7596b4b

bug fix

Files changed (1) hide show

open_vocab_seg/utils/predictor.py CHANGED Viewed

@@ -173,7 +173,7 @@ class SAMVisualizationDemo(object):
         for bbox, mask in zip(bboxes, pred_masks):
             region, _ = crop_with_mask(
                 image,
-                mask,
                 bbox,
                 fill=mask_fill,
             )

         for bbox, mask in zip(bboxes, pred_masks):
             region, _ = crop_with_mask(
                 image,
+                mask.bool(),
                 bbox,
                 fill=mask_fill,
             )