Zhen Ye commited on
Commit
74105c1
·
1 Parent(s): a1ae44f

fix: SAM2 _prep_prompts() signature

Browse files
Files changed (1) hide show
  1. models/segmenters/grounded_sam2.py +1 -1
models/segmenters/grounded_sam2.py CHANGED
@@ -454,7 +454,7 @@ class GroundedSAM2Segmenter(Segmenter):
454
  point_coords=None,
455
  point_labels=None,
456
  box=input_boxes,
457
- mask_input=None,
458
  normalize_coords=True,
459
  )
460
  )
 
454
  point_coords=None,
455
  point_labels=None,
456
  box=input_boxes,
457
+ mask_logits=None,
458
  normalize_coords=True,
459
  )
460
  )