misc

Wandrys-dev · Dec 20, 2022 · 95136b1 · 95136b1
1 parent 82f5222
commit 95136b1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/inst/py/cellpose_utils.py b/inst/py/cellpose_utils.py
@@ -362,7 +362,7 @@ def predict_slice(self, im_dat, dat_slices):
         interm_labels[i] = np.zeros(label_shape, dtype = np.uint32)
 
         for y in model_masks[i]:
-          interm_labels[i] = np.maximum(interm_labels[i], y)
+          interm_labels[i] = np.squeeze(np.maximum(interm_labels[i], y))
 
     # match labels
     if len(model_masks['cyto']) > 0 and len(model_masks['nuc']) > 0:
@@ -390,7 +390,6 @@ def predict_slice(self, im_dat, dat_slices):
     if set({'unmatched', 'cyto'}).issubset(interm_labels.keys()):
       merged_labels = np.maximum(interm_labels['cyto'], interm_labels['unmatched'])
     elif 'cyto' in interm_labels.keys():
-      self.logfile_utils.log('> take cyto')
       merged_labels = interm_labels['cyto']
     elif 'unmatched' in interm_labels.keys():
       merged_labels = interm_labels['unmatched']