Import segmentation instead of alias.

deepcell_toolbox/processing.py

@@ -33,11 +33,11 @@
 import numpy as np
 from scipy import ndimage
 from skimage import morphology
+from skimage import segmentation
 from skimage.feature import peak_local_max
 from skimage.exposure import equalize_adapthist
 from skimage.exposure import rescale_intensity
 from skimage.measure import label
-from skimage.segmentation import watershed as sk_watershed
 
 
 def normalize(image, epsilon=1e-07):
@@ -231,7 +231,7 @@ def watershed(image, min_distance=10, min_size=50, threshold_abs=0.05):
 
     # markers = label(local_maxi)
     markers = ndimage.label(local_maxi)[0]
-    segments = sk_watershed(-distance, markers, mask=labels)
+    segments = segmentation.watershed(-distance, markers, mask=labels)
     results = np.expand_dims(segments, axis=-1)
     results = morphology.remove_small_objects(
         results, min_size=min_size, connectivity=1)