move surface-coordinates postprocessing functions from segmentation t…

…o multistack
fish-quant · Jan 25, 2022 · 398de7f · 398de7f
1 parent 1f96b80
commit 398de7f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/bigfish/plot/plot_images.py b/bigfish/plot/plot_images.py
@@ -9,7 +9,7 @@
 import warnings
 
 import bigfish.stack as stack
-import bigfish.segmentation as segmentation
+import bigfish.multistack as multistack
 
 from .utils import save_plot, get_minmax_values, create_colormap
 
@@ -971,14 +971,14 @@ def plot_cell(ndim, cell_coord=None, nuc_coord=None, rna_coord=None,
                 image = stack.rescale(image, channel_to_stretch=0)
             ax[0].imshow(image)
         if cell_mask is not None:
-            cell_boundaries = segmentation.from_surface_to_boundaries(
+            cell_boundaries = multistack.from_surface_to_boundaries(
                 cell_mask)
             cell_boundaries = np.ma.masked_where(
                 cell_boundaries == 0,
                 cell_boundaries)
             ax[0].imshow(cell_boundaries, cmap=ListedColormap(['red']))
         if nuc_mask is not None:
-            nuc_boundaries = segmentation.from_surface_to_boundaries(nuc_mask)
+            nuc_boundaries = multistack.from_surface_to_boundaries(nuc_mask)
             nuc_boundaries = np.ma.masked_where(
                 nuc_boundaries == 0,
                 nuc_boundaries)