opencv-python : drawMatchesKnn() 总是 return NULL

Question

我想做一个关于标志检测的简单项目。所以我试着按照 OpenCV-Python 关于特征检测的教程。 OpenCV: Feature Matching

我写了如下代码。

ratio = 0.8
logo = cv.imread("T01/CocaCola_logo2.png", cv.IMREAD_GRAYSCALE)
img = cv.imread("T01/CocaCola.png", cv.IMREAD_GRAYSCALE)

orb = cv.ORB_create()
kp_logo, des_logo = orb.detectAndCompute(logo, None)
kp_img, des_img = orb.detectAndCompute(img, None)

FLANN_INDEX_LSH = 6
index_params = dict(algorithm=FLANN_INDEX_LSH,
                    table_number=6,
                    key_size=12,
                    multi_probe_level=1)
search_params = dict(checks=50)
flann = cv.FlannBasedMatcher(index_params, search_params)
match_flann = flann.knnMatch(des_logo, des_img, k=2)

good = []
for p, q in match_flann:
    if p.distance > q.distance*ratio:
        good.append(p)

try:
    img_match = np.empty((max(logo.shape[0], img.shape[0]), logo.shape[1] + img.shape[1], 3), dtype=np.uint8)
    cv.drawMatchesKnn(logo, kp_logo, img, kp_img, good,
                      outImg=img_match, matchColor=None, singlePointColor=(255, 255, 255), flags=2)
    cv.imshow("flann matching", img_match)
    cv.waitKey(0)
except:
    print("...")

此代码无法正常运行。进程总是被 drawMatchesKnn() 错误处理。在我向该函数添加 try-except 之前，进程发生了系统错误：SystemError: returned NULL without setting an error

问题的原因是什么？我试着搜索了很长时间，但对我来说并不容易。

堆栈跟踪：追溯（最近一次通话）：文件 "C:/Users/choib/Desktop/openCVtest3/T01_ORBtest.py"，第 73 行，位于 cv.drawMatchesKnn(标志, kp_logo, img, kp_img, 好, outImg=img_match, matchColor=None, singlePointColor=(255, 255, 255),标志= 2） SystemError: 返回 NULL 没有设置错误

Answer 1

我误解了 drawMatches() 和 drawMatchesKnn()。 reference

drawMatchesKnn()需要获取由DMatch的向量组成的向量作为matches1to2。所以在我的例子中，我修复了如下代码。

案例一：正确使用

img_match = np.empty((max(logo.shape[0], img.shape[0]), logo.shape[1] + img.shape[1], 3), dtype=np.uint8)
cv.drawMatches(logo, kp_logo, img, kp_img, good_flann,
               outImg=img_match, matchColor=None, singlePointColor=(255, 255, 255), flags=2)
cv.imshow("flann matching", img_match)
cv.waitKey(0)

情况 2：使用 drawMatchesKnn()

img_match = np.empty((max(logo.shape[0], img.shape[0]), logo.shape[1] + img.shape[1], 3), dtype=np.uint8)
cv.drawMatchesKnn(logo, kp_logo, img, kp_img, match_flann,
                  outImg=img_match, matchColor=None, singlePointColor=(255, 255, 255), flags=2)
cv.imshow("flann matching", img_match)
cv.waitKey(0)

在这些代码中，good_flann 只是 DMatch 的向量。 match_flann 是向量的向量。所以这些部分的过程没有任何问题。

Answer 2

我收到此错误是因为我试图通过向 drawMatchesKnn 传递一个 DMatch 对象列表来与 drawMatchesKnn 进行匹配。但是，drawMatchesKnn 需要一个列表列表，每个列表包含 DMatch 个对象。

opencv-python : drawMatchesKnn() 总是 return NULL

opencv-python : drawMatchesKnn() always return NULL

python

opencv

computer-vision

feature-detection