使用 Opencv SIFT 时matches1to2 出错

Question

我收到一个错误：

line 33, in <module>
img3 = cv2.drawMatches(img1,kp1,img2,kp2,matches,None,flags=2)
TypeError: Expected cv::DMatch for argument 'matches1to2'

当我尝试执行此代码时。它应该是 SIFT 中的简单图像比较并绘制匹配。我以前让代码适用于 ORB，但是当我将其转换为适用于 SIFT 时，发生了此错误。有人可以帮我解决这个问题吗？代码如下：

#import os
import sys
sys.path.append('/usr/local/lib/python2.7/site-packages')
import cv2
import time
import glob, os
import numpy as np
import matplotlib.pyplot as plt


img1 = cv2.imread("/home/undead/Documents/TestSpectro/test.png",0)
img2 = cv2.imread("/home/undead/Documents/LibrarySpectro/ThankYouAud.png",0)

sift = cv2.xfeatures2d.SIFT_create(3000)

kp1,des1 = sift.detectAndCompute(img1,None)
kp2,des2 = sift.detectAndCompute(img2,None)

bf = cv2.BFMatcher()

matches = bf.knnMatch(des1,des2,k=2)


good = []
for m,n in matches:
    if m.distance < 0.75 * n.distance:
        good.append([m])

#img3 = cv2.drawMatches(img1,kp1,img2,kp2,good,None,flags=2)
img3 = cv2.drawMatches(img1,kp1,img2,kp2,matches,None,flags=2)

plt.imshow(img3),plt.show()

更新 发现问题，有问题的行应该是：

img3 = cv2.drawMatchesKnn(img1,kp1,img2,kp2,matches,None,flags=2)

Answer 1

（除了你的问题）在你的代码中，你应用了比率测试，但你不使用结果。您应该传递“good”而不是“matches”作为参数。

对于您的问题，可以维护两个列表，例如

good = []
good_without_list = []

for m, n in matches:
    if m.distance < 0.75 * n.distance:
        good.append([m])
        good_without_list.append(m)

然后你可以使用

• “cv2.drawMatchesKnn”的“好”列表（在你的情况下）

  knn_image = cv2.drawMatchesKnn(img1, kp1, img2, kp2, good, None, flags=2)
  

• “cv2.drawMatches”的“good_without_list”列表

  img3 = cv2.drawMatches(img1, kp1, img2, kp2, good_without_list, None, **draw_params)