STY: Rework masking based on suggestions by @ahojnnes

This commit is contained in:
Tony S Yu
2012-09-04 09:35:35 -04:00
parent 6c9810c6e9
commit 367e39094c
+6 -4
View File
@@ -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):