Skip to content

Commit b328bb2

Browse files
Update grid search functionality for more flexibility (#1079)
1 parent df4ca88 commit b328bb2

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

micro_sam/evaluation/instance_segmentation.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -229,20 +229,20 @@ def run_instance_segmentation_grid_search(
229229
image = _load_image(image_path, image_key, roi=None if rois is None else rois[i])
230230
gt = _load_image(gt_path, gt_key, roi=None if rois is None else rois[i])
231231

232+
if tiling_window_params is None:
233+
tiling_window_params = {}
234+
232235
if embedding_dir is None:
233236
embedding_path = None
237+
segmenter.initialize(image, **tiling_window_params)
238+
234239
else:
235240
assert predictor is not None
236241
embedding_path = os.path.join(embedding_dir, f"{os.path.splitext(image_name)[0]}.zarr")
237-
238-
if tiling_window_params is None:
239-
tiling_window_params = {}
240-
241-
image_embeddings = util.precompute_image_embeddings(
242-
predictor, image, embedding_path, ndim=2, verbose=verbose_embeddings, **tiling_window_params
243-
)
244-
245-
segmenter.initialize(image, image_embeddings, **tiling_window_params)
242+
image_embeddings = util.precompute_image_embeddings(
243+
predictor, image, embedding_path, ndim=2, verbose=verbose_embeddings, **tiling_window_params
244+
)
245+
segmenter.initialize(image, image_embeddings, **tiling_window_params)
246246

247247
_grid_search_iteration(
248248
segmenter, gs_combinations, gt, image_name,

0 commit comments

Comments
 (0)