Simplify code for coloring boundaries

This commit is contained in:
Juan Nunez-Iglesias
2015-01-12 12:26:41 +11:00
parent 242dbf9b2b
commit 307d4b6137
+3 -3
View File
@@ -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