Skip to content

Commit

Permalink
Fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
anwai98 committed Oct 9, 2024
1 parent 0b48c3f commit ecb1e5d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions micro_sam/evaluation/inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -472,8 +472,8 @@ def run_inference_with_iterative_prompting(
image_paths: List[Union[str, os.PathLike]],
gt_paths: List[Union[str, os.PathLike]],
prediction_dir: Union[str, os.PathLike],
start_with_box_prompt: bool,
embedding_dir: Optional[Union[str, os.PathLike]] = None,
start_with_box_prompt: bool = True,
dilation: int = 5,
batch_size: int = 32,
n_iterations: int = 8,
Expand All @@ -487,8 +487,8 @@ def run_inference_with_iterative_prompting(
image_paths: The image file paths.
gt_paths: The ground-truth segmentation file paths.
prediction_dir: The directory where the predictions from Segment Anything will be saved per iteration.
start_with_box_prompt: Whether to use the first prompt as bounding box or a single point
embedding_dir: The directory where the image embeddings will be saved or are already saved.
start_with_box_prompt: Whether to use the first prompt as bounding box or a single point
dilation: The dilation factor for the radius around the ground-truth object
around which points will not be sampled.
batch_size: The batch size used for batched predictions.
Expand Down

0 comments on commit ecb1e5d

Please sign in to comment.