def mspp(self): file_path1= filedialog.askopenfilename() file_path2 = filedialog.askopenfilename() queryImage = cv2.imread(file_path1, 0) trainingImage = cv2.imread(file_path2, 0) # 只使用SIFT 或 SURF 检测角点 sift = cv2.SIFT_create() # sift = cv2.xfeatures2d.SURF_create(float(4000)) kp1, des1 = sift.detectAndCompute(queryImage, None) kp2, des2 = sift.detectAndCompute(trainingImage, None) # 设置FLANN匹配器参数 # algorithm设置可参考https://docs.opencv.org/3.1.0/dc/d8c/namespacecvflann.html indexParams = dict(algorithm=0, trees=5) searchParams = dict(checks=50) # 定义FLANN匹配器 flann = cv2.FlannBasedMatcher(indexParams, searchParams) # 使用 KNN 算法实现匹配 matches = flann.knnMatch(des1, des2, k=2) # 根据matches生成相同长度的matchesMask列表，列表元素为[0,0] matchesMask = [[0, 0] for i in range(len(matches))] # 去除错误匹配 for i, (m, n) in enumerate(matches): if m.distance < 0.7 * n.distance: matchesMask[i] = [1, 0] # 将图像显示 # matchColor是两图的匹配连接线，连接线与matchesMask相关 # singlePointColor是勾画关键点 drawParams = dict(matchColor=(0, 255, 0), singlePointColor=(255, 0, 0), matchesMask=matchesMask, flags=0) resultImage = cv2.drawMatchesKnn(queryImage, kp1, trainingImage, kp2, matches, None, **drawParams) plt.imshow(resultImage, ), plt.show()

这段代码实现了图像特征点匹配的功能，使用的是 OpenCV 库。首先通过 filedialog 选择两个要进行匹配的图像，然后使用 SIFT 算法检测两个图像的特征点。接着，使用 FLANN 算法对两张图像的特征点进行匹配，并去除错误匹配。最后，将匹配结果可视化并显示出来。其中，kp1、des1、kp2、des2 分别表示两张图像检测出的特征点和对应的特征描述子向量。matches 表示 FLANN 算法匹配得到的特征点对，matchesMask 表示筛选后的匹配对应的掩码，resultImage 表示可视化后的匹配结果图像。