diff --git a/doc/examples/plot_orb_matching.py b/doc/examples/plot_orb_matching.py index 7d1119ab..78409dd1 100644 --- a/doc/examples/plot_orb_matching.py +++ b/doc/examples/plot_orb_matching.py @@ -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() +