OpenCV功能匹配多个图像

发布于 2021-01-29 15:06:16

如何使用FLANN优化许多图片的SIFT功能匹配?

我有一个来自Python
OpenCV文档的工作示例。但是,这是将一个图像与另一个图像进行比较,而且速度很慢。我需要它来搜索一系列图像(几千个)中匹配的特征,并且我需要它更快。

我目前的想法:

  1. 浏览所有图像并保存功能。怎么样?
  2. 将相机的图像与上述基础进行比较,然后找到正确的图像。怎么样?
  3. 给我结果,匹配图像或其他东西。

http://docs.opencv.org/trunk/doc/py_tutorials/py_feature2d/py_feature_homography/py_feature_homography.html

import sys#仅用于调试
将numpy导入为np
导入cv2
从matplotlib导入pyplot作为plt

MIN_MATCH_COUNT = 10

img1 = cv2.imread('image.jpg',0)#queryImage
img2 = cv2.imread('target.jpg',0)#trainImage

#启动SIFT检测器
筛选= cv2.SIFT()

#使用SIFT查找关键点和描述符
kp1,des1 = sift.detectAndCompute(img1,None)
kp2,des2 = sift.detectAndCompute(img2,None)

FLANN_INDEX_KDTREE = 0
index_params = dict(算法= FLANN_INDEX_KDTREE,树= 5)
search_params = dict(checks = 50)

flann = cv2.FlannBasedMatcher(index_params,search_params)

匹配= flann.knnMatch(des1,des2,k = 2)

#根据Lowe的比率测试存储所有符合条件的匹配项。
好= []
对于在比赛中的m,n:
    如果m.distance MIN_MATCH_COUNT:
    src_pts = np.float32([k p1 [m.queryIdx] .pt for m in good])。reshape(-1,1,2)
    dst_pts = np.float32([k p2 [m.trainIdx] .pt for m in good])。reshape(-1,1,2)

    M,遮罩= cv2.findHomography(src_pts,dst_pts,cv2.RANSAC,5.0)
    matchsMask = mask.ravel()。tolist()

    h,w = img1.shape
    pts = np.float32([[0,0],[0,h-1],[w-1,h-1],[w-1,0]])。reshape(-1,1,2)
    dst = cv2.perspectiveTransform(pts,M)

    img2 = cv2.polylines(img2,[np.int32(dst)],True,255,3,cv2.LINE_AA)

其他:
    打印“找不到足够的匹配项-%d /%d”%(len(good),MIN_MATCH_COUNT)
    matchsMask =无

draw_params = dict(matchColor =(0,255,0),#用绿色绘制匹配项
                   singlePointColor =无,
                   matchesMask = matchesMask,#只画内线
                   标志= 2)

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

plt.imshow(img3,'灰色'),plt.show()

更新

在尝试了许多事情之后,我现在可能更接近解决方案了。我希望可以建立索引,然后像这样搜索它:

flann_params = dict(算法= 1,树木= 4)
flann = cv2.flann_Index(npArray,flann_params)
idx,dist = flann.knnSearch(queryDes,1,params = {})

但是我仍然没有设法为flann_Index参数建立一个可接受的npArray。

遍历所有图像作为图像:
  npArray.append(sift.detectAndCompute(image,None))
npArray = np.array(npArray)
关注者
0
被浏览
56
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    我从来没有在Python中解决过这个问题,但是我将环境切换到了C ++,在那里您可以获得更多的OpenCV示例,而不必使用包装较少的文档。

    我在多个文件中存在匹配问题的示例可以在这里找到:https
    :
    //github.com/Itseez/opencv/blob/2.4/samples/cpp/matching_to_many_images.cpp



知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看