wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-12 08:54:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Allocate less memory, if possible.

Browse Source
This commit is contained in:
Stefan van der Walt
2013-10-14 13:02:38 +02:00
parent 89d703ca46
commit d5e2ab0b13
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/morphology/binary.py
+4 -2
View File
@@ -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: