Skip to content

Commit e4f7f00

Browse files
authored
fix custom model use with Cellpose-SAM (#285)
1 parent 9398103 commit e4f7f00

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

active_plugins/runcellpose.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -872,12 +872,13 @@ def run(self, workspace):
872872

873873
elif self.cellpose_version.value == 'v4':
874874
assert int(self.cellpose_ver[0])==4, "Cellpose version selected in RunCellpose module doesn't match version in Python"
875+
LOGGER.info(f"Loading new model: {self.mode.value}")
875876
if self.mode.value == 'custom':
876877
model_file, model_directory, model_path = get_custom_model_vars(self)
877-
model_params = (self.mode.value, self.use_gpu.value)
878-
LOGGER.info(f"Loading new model: {self.mode.value}")
879-
self.current_model = models.CellposeModel(gpu=self.use_gpu.value)
880-
self.current_model_params = model_params
878+
self.current_model = models.CellposeModel(
879+
pretrained_model=model_path, gpu=self.use_gpu.value)
880+
else:
881+
self.current_model = models.CellposeModel(gpu=self.use_gpu.value)
881882

882883
if self.use_gpu.value:
883884
try:

0 commit comments

Comments
 (0)