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

Add bandwidth parameter to gabor_kernel

Browse Source 
Note that this changes the API of `gabor_kernel` and `gabor_filter`: The input parameters are rearranged and positional arguments are changed to keyword args.
This commit is contained in:
Tony S Yu
2013-04-06 21:18:53 -05:00
parent 7d407fc778
commit 2bf178ceb2
2 changed files with 59 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/filter/tests/test_gabor.py
+25 -11
View File
@@ -2,7 +2,7 @@ import numpy as np
from numpy.testing import (assert_equal, assert_almost_equal,
assert_array_almost_equal)
from skimage.filter import gabor_kernel, gabor_filter
from skimage.filter._gabor import gabor_kernel, gabor_filter, _sigma_prefactor
def test_gabor_kernel_size():
@@ -12,21 +12,35 @@ def test_gabor_kernel_size():
size_x = sigma_x * 6 + 1
size_y = sigma_y * 6 + 1
theta = 0
kernel = gabor_kernel(sigma_x, sigma_y, 0, theta)
kernel = gabor_kernel(0, theta=0, sigma_x=sigma_x, sigma_y=sigma_y)
assert_equal(kernel.shape, (size_y, size_x))
theta = np.pi / 2
kernel = gabor_kernel(sigma_x, sigma_y, 0, theta)
kernel = gabor_kernel(0, theta=np.pi/2, sigma_x=sigma_x, sigma_y=sigma_y)
assert_equal(kernel.shape, (size_x, size_y))
def test_gabor_kernel_bandwidth():
kernel = gabor_kernel(1, bandwidth=1)
assert_equal(kernel.shape, (5, 5))
kernel = gabor_kernel(1, bandwidth=0.5)
assert_equal(kernel.shape, (9, 9))
kernel = gabor_kernel(0.5, bandwidth=1)
assert_equal(kernel.shape, (9, 9))
def test_sigma_prefactor():
assert_almost_equal(_sigma_prefactor(1), 0.56, 2)
assert_almost_equal(_sigma_prefactor(0.5), 1.09, 2)
def test_gabor_kernel_sum():
for sigma_x in range(1, 10, 2):
for sigma_y in range(1, 10, 2):
for frequency in range(0, 10, 2):
kernel = gabor_kernel(sigma_x, sigma_y, frequency+0.1, 0)
kernel = gabor_kernel(frequency+0.1, theta=0,
sigma_x=sigma_x, sigma_y=sigma_y)
# make sure gaussian distribution is covered nearly 100%
assert_almost_equal(np.abs(kernel).sum(), 1, 2)
@@ -36,17 +50,17 @@ def test_gabor_kernel_theta():
for sigma_y in range(1, 10, 2):
for frequency in range(0, 10, 2):
for theta in range(0, 10, 2):
kernel0 = gabor_kernel(sigma_x, sigma_y, frequency+0.1,
theta)
kernel180 = gabor_kernel(sigma_x, sigma_y, frequency,
theta+np.pi)
kernel0 = gabor_kernel(frequency+0.1, theta=theta,
sigma_x=sigma_x, sigma_y=sigma_y)
kernel180 = gabor_kernel(frequency, theta=theta+np.pi,
sigma_x=sigma_x, sigma_y=sigma_y)
assert_array_almost_equal(np.abs(kernel0),
np.abs(kernel180))
def test_gabor_filter():
real, imag = gabor_filter(np.random.random((100, 100)), 1, 1, 1, 1)
real, imag = gabor_filter(np.random.random((100, 100)), 1)
if __name__ == "__main__":