From d5e2ab0b135dc959ff07c5eddf67af808d8c908e Mon Sep 17 00:00:00 2001 From: Stefan van der Walt Date: Mon, 14 Oct 2013 13:02:38 +0200 Subject: [PATCH] Allocate less memory, if possible. --- skimage/morphology/binary.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/skimage/morphology/binary.py b/skimage/morphology/binary.py index fe91f148..6bd5d234 100644 --- a/skimage/morphology/binary.py +++ b/skimage/morphology/binary.py @@ -34,10 +34,11 @@ def binary_erosion(image, selem, out=None): if selem_sum <= 255: binary = (image > 0).view(np.uint8) + conv = np.empty_like(image, dtype=np.uint8) else: binary = (image > 0).astype(np.intp) + conv = np.empty_like(image, dtype=np.intp) - conv = np.empty_like(image, dtype=np.intp) ndimage.convolve(binary, selem, mode='constant', cval=1, output=conv) if out is None: @@ -75,10 +76,11 @@ def binary_dilation(image, selem, out=None): selem = (selem != 0) if np.sum(selem) <= 255: binary = (image > 0).view(np.uint8) + conv = np.empty_like(image, dtype=np.uint8) else: binary = (image > 0).astype(np.intp) + conv = np.empty_like(image, dtype=np.intp) - conv = np.empty_like(image, dtype=np.intp) ndimage.convolve(binary, selem, mode='constant', cval=0, output=conv) if out is None: