From 3d8d8791d9ad584eac18a2cdf25fb396f73877a4 Mon Sep 17 00:00:00 2001 From: Constantin Pape Date: Wed, 3 Jul 2024 22:45:11 +0200 Subject: [PATCH] Fix bug in precompute for 3d data (#649) --- micro_sam/precompute_state.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/micro_sam/precompute_state.py b/micro_sam/precompute_state.py index e2ddc1ac..d07ea1bc 100644 --- a/micro_sam/precompute_state.py +++ b/micro_sam/precompute_state.py @@ -68,7 +68,7 @@ def cache_amg_state( if verbose: print("Precomputing the state for instance segmentation.") - amg.initialize(raw, image_embeddings=image_embeddings, verbose=verbose, i=i) + amg.initialize(raw if i is None else raw[i], image_embeddings=image_embeddings, verbose=verbose, i=i) amg_state = amg.get_state() # put all state onto the cpu so that the state can be deserialized without a gpu