fixing denoise chan2 bug

MouseLand · Feb 21, 2024 · af4f8e9 · af4f8e9
1 parent e0645b8
commit af4f8e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cellpose/denoise.py b/cellpose/denoise.py
@@ -643,7 +643,7 @@ def __init__(self, gpu=False, pretrained_model=False, nchan=1, model_type=None,
                 f">>>> model diam_mean = {self.diam_mean: .3f} (ROIs rescaled to this size during training)"
             )
             if chan2 and builtin:
-                chan2_path = model_path(self.pretrained_model.stem.split("_")[:-1] + "_nuclei")
+                chan2_path = model_path(os.path.split(self.pretrained_model)[-1].split("_")[0] + "_nuclei")
                 print(f"loading model for chan2: {os.path.split(str(chan2_path)[-1])}")
                 self.net_chan2 = CPnet(self.nbase, self.nclasses, sz=3,
                                        mkldnn=self.mkldnn, max_pool=True,