Skip to content

Commit 3672dce

Browse files
committed
fix: make sure inference state only read once
1 parent df4ab57 commit 3672dce

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

saber/segmenters/tomo.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,8 +92,8 @@ def segment(
9292
# If A Mask is Found, Follow to 3D Segmentation Propagation
9393

9494
# Initialize Video Predictor
95-
# if self.inference_state is None:
96-
self.inference_state = self.video_predictor.create_inference_state_from_tomogram(vol)
95+
if self.inference_state is None:
96+
self.inference_state = self.video_predictor.create_inference_state_from_tomogram(vol)
9797

9898
# Set up a dictionary to capture the object score logits from the mask decoder.
9999
# The keys will be frame indices and the values will be a list of score arrays from that frame.

0 commit comments

Comments
 (0)