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

Fix match tests

Browse Source
This commit is contained in:
Johannes Schönberger
2014-01-19 10:33:29 -05:00
parent 336f0ca266
commit 6307e24372
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/feature/tests/test_match.py
+10 -10
View File
@@ -34,15 +34,15 @@ def test_binary_descriptors_lena_rotation_crosscheck_false():
tform = tf.SimilarityTransform(scale=1, rotation=0.15, translation=(0, 0))
rotated_img = tf.warp(img, tform)
descriptor = BRIEF(descriptor_size=512)
extractor = BRIEF(descriptor_size=512)
keypoints1 = corner_peaks(corner_harris(img), min_distance=5)
descriptors1, mask1 = descriptor.extract(img, keypoints1)
keypoints1 = keypoints1[mask1]
extractor.extract(img, keypoints1)
descriptors1 = extractor.descriptors_
keypoints2 = corner_peaks(corner_harris(rotated_img), min_distance=5)
descriptors2, mask2 = descriptor.extract(rotated_img, keypoints2)
keypoints2 = keypoints1[mask2]
extractor.extract(rotated_img, keypoints2)
descriptors2 = extractor.descriptors_
matches = match_descriptors(descriptors1, descriptors2, threshold=0.13,
cross_check=False)
@@ -68,15 +68,15 @@ def test_binary_descriptors_lena_rotation_crosscheck_true():
tform = tf.SimilarityTransform(scale=1, rotation=0.15, translation=(0, 0))
rotated_img = tf.warp(img, tform)
descriptor = BRIEF(descriptor_size=512)
extractor = BRIEF(descriptor_size=512)
keypoints1 = corner_peaks(corner_harris(img), min_distance=5)
descriptors1, mask1 = descriptor.extract(img, keypoints1)
keypoints1 = keypoints1[mask1]
extractor.extract(img, keypoints1)
descriptors1 = extractor.descriptors_
keypoints2 = corner_peaks(corner_harris(rotated_img), min_distance=5)
descriptors2, mask2 = descriptor.extract(rotated_img, keypoints2)
keypoints2 = keypoints1[mask2]
extractor.extract(rotated_img, keypoints2)
descriptors2 = extractor.descriptors_
matches = match_descriptors(descriptors1, descriptors2, threshold=0.13,
cross_check=True)