Skip to content

Commit fd4ef37

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 5e8c079 commit fd4ef37

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

computer_vision/image_segmentation_sam2.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -221,7 +221,9 @@ def demonstrate_point_prompt(predictor, image):
221221
return masks, scores, logits
222222

223223

224-
def demonstrate_multiple_points(predictor, image, previous_masks, previous_scores, previous_logits):
224+
def demonstrate_multiple_points(
225+
predictor, image, previous_masks, previous_scores, previous_logits
226+
):
225227
"""Demonstrate segmentation using multiple point prompts"""
226228
print("=== Multiple Points Prompt ===")
227229

@@ -358,7 +360,7 @@ def demonstrate_batched_images(predictor):
358360

359361
# Process batch
360362
predictor.set_image_batch(img_batch)
361-
masks_batch, _ , _ = predictor.predict_batch(
363+
masks_batch, _, _ = predictor.predict_batch(
362364
None, None, box_batch=boxes_batch, multimask_output=False
363365
)
364366

0 commit comments

Comments
 (0)