From 307d4b6137962910c7fffdad32e10ee37b3324e2 Mon Sep 17 00:00:00 2001 From: Juan Nunez-Iglesias Date: Mon, 12 Jan 2015 12:26:41 +1100 Subject: [PATCH] Simplify code for coloring boundaries --- skimage/segmentation/boundaries.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/skimage/segmentation/boundaries.py b/skimage/segmentation/boundaries.py index c637dd5f..a2c5f3fb 100644 --- a/skimage/segmentation/boundaries.py +++ b/skimage/segmentation/boundaries.py @@ -208,7 +208,7 @@ def mark_boundaries(image, label_img, color=(1, 1, 0), boundaries = find_boundaries(label_img, mode=mode, background=background_label) if outline_color is not None: - outer_boundaries = dilation(boundaries.astype(np.uint8), square(3)) - marked[outer_boundaries != 0, :] = np.array(outline_color) - marked[boundaries, :] = np.array(color) + outlines = dilation(boundaries, square(3)) + marked[outlines] = outline_color + marked[boundaries] = color return marked