Fix for predefined output array

This commit is contained in:
Johannes Schönberger
2012-09-02 13:41:25 +02:00
parent c47e230847
commit aa08e8a559
+8 -4
View File
@@ -29,8 +29,10 @@ def binary_erosion(image, selem, out=None):
"""
out = ndimage.convolve(image > 0, selem, output=out,
mode='constant', cval=1)
conv = ndimage.convolve(image > 0, selem, output=out,
mode='constant', cval=1)
if conv is not None:
out = conv
return np.equal(out, np.sum(selem), out=out)
@@ -62,8 +64,10 @@ def binary_dilation(image, selem, out=None):
"""
out = ndimage.convolve(image > 0, selem, output=out,
mode='constant', cval=1)
conv = ndimage.convolve(image > 0, selem, output=out,
mode='constant', cval=1)
if conv is not None:
out = conv
return np.not_equal(out, 0, out=out)