mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-27 19:48:43 +08:00
Johannes Schönberger
81 lines
3.1 KiB
Python
81 lines
3.1 KiB
Python
import numpy as np
|
|
from numpy.testing import assert_array_equal, assert_raises
|
|
from skimage import data
|
|
from skimage import transform as tf
|
|
from skimage.color import rgb2gray
|
|
from skimage.feature import BRIEF, corner_peaks, corner_harris
|
|
from skimage._shared.testing import test_parallel
|
|
|
|
|
|
def test_color_image_unsupported_error():
|
|
"""Brief descriptors can be evaluated on gray-scale images only."""
|
|
img = np.zeros((20, 20, 3))
|
|
keypoints = np.asarray([[7, 5], [11, 13]])
|
|
assert_raises(ValueError, BRIEF().extract, img, keypoints)
|
|
|
|
|
|
def test_normal_mode():
|
|
"""Verify the computed BRIEF descriptors with expected for normal mode."""
|
|
img = data.coins()
|
|
|
|
keypoints = corner_peaks(corner_harris(img), min_distance=5,
|
|
threshold_abs=0, threshold_rel=0.1)
|
|
|
|
extractor = BRIEF(descriptor_size=8, sigma=2)
|
|
|
|
extractor.extract(img, keypoints[:8])
|
|
|
|
expected = np.array([[False, True, False, False, True, False, True, False],
|
|
[ True, False, True, True, False, True, False, False],
|
|
[ True, False, False, True, False, True, False, True],
|
|
[ True, True, True, True, False, True, False, True],
|
|
[ True, True, True, False, False, True, True, True],
|
|
[False, False, False, False, True, False, False, False],
|
|
[False, True, False, False, True, False, True, False],
|
|
[False, False, False, False, False, False, False, False]], dtype=bool)
|
|
|
|
assert_array_equal(extractor.descriptors, expected)
|
|
|
|
|
|
def test_uniform_mode():
|
|
"""Verify the computed BRIEF descriptors with expected for uniform mode."""
|
|
img = data.coins()
|
|
|
|
keypoints = corner_peaks(corner_harris(img), min_distance=5,
|
|
threshold_abs=0, threshold_rel=0.1)
|
|
|
|
extractor = BRIEF(descriptor_size=8, sigma=2, mode='uniform')
|
|
|
|
extractor.extract(img, keypoints[:8])
|
|
|
|
expected = np.array([[False, False, False, True, True, True, False, False],
|
|
[ True, True, True, False, True, False, False, True],
|
|
[ True, True, True, False, True, True, False, True],
|
|
[ True, True, True, True, False, True, False, True],
|
|
[ True, True, True, True, True, True, False, False],
|
|
[ True, True, True, True, True, True, True, True],
|
|
[False, False, False, True, True, True, True, True],
|
|
[False, True, False, True, False, True, True, True]], dtype=bool)
|
|
|
|
assert_array_equal(extractor.descriptors, expected)
|
|
|
|
|
|
def test_unsupported_mode():
|
|
assert_raises(ValueError, BRIEF, mode='foobar')
|
|
|
|
|
|
def test_border():
|
|
img = np.zeros((100, 100))
|
|
keypoints = np.array([[1, 1], [20, 20], [50, 50], [80, 80]])
|
|
|
|
extractor = BRIEF(patch_size=41)
|
|
extractor.extract(img, keypoints)
|
|
|
|
assert extractor.descriptors.shape[0] == 3
|
|
assert_array_equal(extractor.mask, (False, True, True, True))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from numpy import testing
|
|
testing.run_module_suite()
|