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

Making matching graphics more clear

Browse Source
This commit is contained in:
Ankit Agrawal
2013-09-26 17:58:37 +05:30
committed by Johannes Schönberger
parent a93fe9dbb9
commit 11403f9ce3
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/examples/plot_orb_matching.py
+10 -8
View File
@@ -9,7 +9,7 @@ import matplotlib.pyplot as plt
rotate = 0.5
translate = (-100, -200)
scaling = (1.5, 1.5)
match_threshold = 0.40
match_threshold = 0.25
match_cross_check = True
img_color = data.lena()
@@ -18,13 +18,13 @@ transformed_img_color = tf.warp(img_color, tform)
img = rgb2gray(img_color)
transformed_img = rgb2gray(transformed_img_color)
keypoints1, orientations1, scales1 = keypoints_orb(img, n_keypoints=250)
keypoints1, orientations1, scales1 = keypoints_orb(img, n_keypoints=500)
keypoints1.shape
descriptors1, keypoints1 = descriptor_orb(img, keypoints1, orientations1, scales1)
keypoints1.shape
descriptors1.shape
keypoints2, orientations2, scales2 = keypoints_orb(transformed_img, n_keypoints=250)
keypoints2, orientations2, scales2 = keypoints_orb(transformed_img, n_keypoints=500)
keypoints2.shape
descriptors2, keypoints2 = descriptor_orb(transformed_img, keypoints2, orientations2, scales2)
keypoints2.shape
@@ -38,8 +38,8 @@ matched_keypoints.shape
# Plotting the matched correspondences in both the images using matplotlib
src = matched_keypoints[:, 0, :]
dst = matched_keypoints[:, 1, :]
src_scale = 10 * (scales1[mask1] + 1) ** 2
dst_scale = 10 * (scales2[mask2] + 1) ** 2
src_scale = 10 * (scales1[mask1] + 1) ** 1.5
dst_scale = 10 * (scales2[mask2] + 1) ** 1.5
img_combined = np.concatenate((img_as_float(img_color), img_as_float(transformed_img_color)), axis=1)
offset = img.shape
@@ -53,8 +53,10 @@ ax.axis((0, 2 * offset[1], offset[0], 0))
ax.set_title('Matched correspondences : Rotation = %f; Scale = %s; Translation = %s; threshold = %f; cross_check = %r' % (rotate, scaling, translate, match_threshold, match_cross_check))
for m in range(len(src)):
ax.plot((src[m, 1], dst[m, 1] + offset[1]), (src[m, 0], dst[m, 0]), '-', color='g')
ax.scatter(src[m, 1], src[m, 0], src_scale[m], facecolors='none', edgecolors='b')
ax.scatter(dst[m, 1] + offset[1], dst[m, 0], dst_scale[m], facecolors='none', edgecolors='b')
c = np.random.rand(3,1)
ax.plot((src[m, 1], dst[m, 1] + offset[1]), (src[m, 0], dst[m, 0]), '-', color=c)
ax.scatter(src[m, 1], src[m, 0], src_scale[m], facecolors='none', edgecolors=c)
ax.scatter(dst[m, 1] + offset[1], dst[m, 0], dst_scale[m], facecolors='none', edgecolors=c)
plt.show()