注意
转到末尾下载完整的示例代码。或通过 Binder 在您的浏览器中运行此示例
SIFT 特征检测器和描述符提取器#
此示例演示了 SIFT 特征检测及其描述算法。
尺度不变特征变换 (SIFT) [1] 于 1999 年发布,至今仍然是最流行的特征检测器之一,因为它承诺“对图像缩放、平移和旋转不变,并且部分对光照变化和仿射或 3D 投影不变”[2]。它最大的缺点是它的运行时间,据说比 ORB 慢“两个数量级” [3],这使得它不适合实时应用程序。
参考文献#
import matplotlib.pyplot as plt
from skimage import data
from skimage import transform
from skimage.color import rgb2gray
from skimage.feature import match_descriptors, plot_matched_features, SIFT
img1 = rgb2gray(data.astronaut())
img2 = transform.rotate(img1, 180)
tform = transform.AffineTransform(scale=(1.3, 1.1), rotation=0.5, translation=(0, -200))
img3 = transform.warp(img1, tform)
descriptor_extractor = SIFT()
descriptor_extractor.detect_and_extract(img1)
keypoints1 = descriptor_extractor.keypoints
descriptors1 = descriptor_extractor.descriptors
descriptor_extractor.detect_and_extract(img2)
keypoints2 = descriptor_extractor.keypoints
descriptors2 = descriptor_extractor.descriptors
descriptor_extractor.detect_and_extract(img3)
keypoints3 = descriptor_extractor.keypoints
descriptors3 = descriptor_extractor.descriptors
matches12 = match_descriptors(
descriptors1, descriptors2, max_ratio=0.6, cross_check=True
)
matches13 = match_descriptors(
descriptors1, descriptors3, max_ratio=0.6, cross_check=True
)
fig, ax = plt.subplots(nrows=2, ncols=2, figsize=(11, 8))
plt.gray()
plot_matched_features(
img1,
img2,
keypoints0=keypoints1,
keypoints1=keypoints2,
matches=matches12,
ax=ax[0, 0],
)
ax[0, 0].axis('off')
ax[0, 0].set_title("Original Image vs. Flipped Image\n" "(all keypoints and matches)")
plot_matched_features(
img1,
img3,
keypoints0=keypoints1,
keypoints1=keypoints3,
matches=matches13,
ax=ax[1, 0],
)
ax[1, 0].axis('off')
ax[1, 0].set_title(
"Original Image vs. Transformed Image\n" "(all keypoints and matches)"
)
plot_matched_features(
img1,
img2,
keypoints0=keypoints1,
keypoints1=keypoints2,
matches=matches12[::15],
ax=ax[0, 1],
only_matches=True,
)
ax[0, 1].axis('off')
ax[0, 1].set_title(
"Original Image vs. Flipped Image\n" "(subset of matches for visibility)"
)
plot_matched_features(
img1,
img3,
keypoints0=keypoints1,
keypoints1=keypoints3,
matches=matches13[::15],
ax=ax[1, 1],
only_matches=True,
)
ax[1, 1].axis('off')
ax[1, 1].set_title(
"Original Image vs. Transformed Image\n" "(subset of matches for visibility)"
)
plt.tight_layout()
plt.show()
脚本总运行时间:(0 分钟 6.490 秒)