Skip to content

Commit 648ed53

Browse files
authored
Update image_segmentation_sam2.py
1 parent 6b33680 commit 648ed53

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

computer_vision/image_segmentation_sam2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -222,7 +222,7 @@ def demonstrate_point_prompt(predictor, image):
222222

223223

224224
def demonstrate_multiple_points(
225-
predictor, image, previous_masks, previous_scores, previous_logits
225+
predictor, image, previous_scores, previous_logits
226226
):
227227
"""Demonstrate segmentation using multiple point prompts"""
228228
print("=== Multiple Points Prompt ===")
@@ -406,7 +406,7 @@ def main():
406406

407407
# Run demonstrations
408408
masks, scores, logits = demonstrate_point_prompt(predictor, image)
409-
demonstrate_multiple_points(predictor, image, masks, scores, logits)
409+
demonstrate_multiple_points(predictor, image, scores, logits)
410410
demonstrate_box_prompt(predictor, image)
411411
demonstrate_combined_prompts(predictor, image)
412412
demonstrate_batched_prompts(predictor, image)

0 commit comments

Comments
 (0)