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

CLN: Improve performance of disk and diamond selems

Browse Source
This commit is contained in:
Pratap Vardhan
2014-10-07 17:55:44 +05:30
parent 669b0bf311
commit 39267d103c
1 changed files with 7 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/morphology/selem.py
+7 -9
View File
@@ -85,10 +85,10 @@ def diamond(radius, dtype=np.uint8):
The structuring element where elements of the neighborhood
are 1 and 0 otherwise.
"""
half = radius
(I, J) = np.meshgrid(range(0, radius * 2 + 1), range(0, radius * 2 + 1))
s = np.abs(I - half) + np.abs(J - half)
return np.array(s <= radius, dtype=dtype)
L = np.arange(0, radius * 2 + 1)
I, J = np.meshgrid(L, L)
return np.array(np.abs(I - radius) + np.abs(J - radius) <= radius,
dtype=dtype)
def disk(radius, dtype=np.uint8):
@@ -113,11 +113,9 @@ def disk(radius, dtype=np.uint8):
The structuring element where elements of the neighborhood
are 1 and 0 otherwise.
"""
L = np.linspace(-radius, radius, 2 * radius + 1)
(X, Y) = np.meshgrid(L, L)
s = X**2
s += Y**2
return np.array(s <= radius * radius, dtype=dtype)
L = np.arange(-radius, radius + 1)
X, Y = np.meshgrid(L, L)
return np.array((X ** 2 + Y ** 2) <= radius ** 2, dtype=np.uint8)
def cube(width, dtype=np.uint8):