From 367e39094c3658e0e63f84f8c334934c4967a77b Mon Sep 17 00:00:00 2001 From: Tony S Yu Date: Tue, 4 Sep 2012 09:35:35 -0400 Subject: [PATCH] STY: Rework masking based on suggestions by @ahojnnes --- skimage/filter/edges.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/skimage/filter/edges.py b/skimage/filter/edges.py index 1568886c..134aa796 100644 --- a/skimage/filter/edges.py +++ b/skimage/filter/edges.py @@ -23,12 +23,14 @@ def _mask_filter_result(result, mask): affect values in the result. """ if mask is None: - mask = np.zeros(result.shape, bool) - mask[1:-1, 1:-1] = True + result[0, :] = 0 + result[-1, :] = 0 + result[:, 0] = 0 + result[:, -1] = 0 + return result else: mask = binary_erosion(mask, EROSION_SELEM, border_value=0) - result[mask == False] = 0 - return result + return result * mask def sobel(image, mask=None):