mirror of
https://github.com/wassname/scikit-image.git
synced 2026-08-02 13:03:48 +08:00
78 lines
3.0 KiB
Python
78 lines
3.0 KiB
Python
import numpy as np
|
|
from numpy.testing import assert_array_equal, assert_raises
|
|
from skimage.data import moon
|
|
from skimage.feature import censure_keypoints
|
|
|
|
|
|
def test_censure_keypoints_color_image_unsupported_error():
|
|
"""Censure keypoints can be extracted from gray-scale images only."""
|
|
img = np.zeros((20, 20, 3))
|
|
assert_raises(ValueError, censure_keypoints, img)
|
|
|
|
|
|
def test_censure_keypoints_moon_image_DoB():
|
|
"""Verify the actual Censure keypoints and their corresponding scale with
|
|
the expected values for DoB filter."""
|
|
img = moon()
|
|
actual_kp_DoB, actual_scale = censure_keypoints(img, 7, 'DoB', 0.15)
|
|
expected_kp_DoB = np.array([[ 4, 507],
|
|
[ 8, 503],
|
|
[ 12, 499],
|
|
[ 21, 497],
|
|
[ 36, 46],
|
|
[119, 350],
|
|
[185, 177],
|
|
[287, 250],
|
|
[357, 239],
|
|
[463, 116],
|
|
[464, 132],
|
|
[467, 260]])
|
|
expected_scale = np.array([2, 4, 6, 3, 4, 4, 2, 2, 3, 2, 2, 2])
|
|
|
|
assert_array_equal(expected_kp_DoB, actual_kp_DoB)
|
|
assert_array_equal(expected_scale, actual_scale)
|
|
|
|
|
|
def test_censure_keypoints_moon_image_Octagon():
|
|
"""Verify the actual Censure keypoints and their corresponding scale with
|
|
the expected values for Octagon filter."""
|
|
img = moon()
|
|
actual_kp_Octagon, actual_scale = censure_keypoints(img, 7, 'Octagon', 0.15)
|
|
expected_kp_Octagon = np.array([[ 21, 496],
|
|
[ 35, 46],
|
|
[287, 250],
|
|
[356, 239],
|
|
[463, 116]])
|
|
|
|
expected_scale = np.array([3, 4, 2, 2, 2], dtype=np.intp)
|
|
|
|
assert_array_equal(expected_kp_Octagon, actual_kp_Octagon)
|
|
assert_array_equal(expected_scale, actual_scale)
|
|
|
|
|
|
def test_censure_keypoints_moon_image_STAR():
|
|
"""Verify the actual Censure keypoints and their corresponding scale with
|
|
the expected values for STAR filter."""
|
|
img = moon()
|
|
actual_kp_STAR, actual_scale = censure_keypoints(img, 7, 'STAR', 0.15)
|
|
expected_kp_STAR = np.array([[ 21, 497],
|
|
[ 36, 46],
|
|
[117, 356],
|
|
[185, 177],
|
|
[260, 227],
|
|
[287, 250],
|
|
[357, 239],
|
|
[451, 281],
|
|
[463, 116],
|
|
[467, 260]])
|
|
|
|
expected_scale = np.array([3, 3, 6, 2, 3, 2, 3, 5, 2, 2], dtype=np.intp)
|
|
|
|
assert_array_equal(expected_kp_STAR, actual_kp_STAR)
|
|
assert_array_equal(expected_scale, actual_scale)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from numpy import testing
|
|
testing.run_module_suite()
|