最近用opencv-python 做一个东西,用到了cv2.flann_Index
,用下式建立索引:
flann = cv2.flann_Index(data, dict(algorithm=1, trees=5)) # kd-tree
然后用 flann.knnSearch()
方法进行knn搜索,搜索结果错误很严重。
错误原因:
cv2.flann_Index
建立索引之后,原始数据一定不能删除,否则查询结果会变得莫名其妙。
cv2.flann_Index
的正确打开方式:
img1 = cv2.imread('/home/lilei/data/frame1.jpg')
img2 = cv2.imread('/home/lilei/data/frame1.jpg')
sift = cv2.xfeatures2d.SIFT_create()
kpt1, des1 = sift.detectAndCompute(img1, None)
kpt2, des2 = sift.detectAndCompute(img2, None)
flann = cv2.flann_Index(des1, dict(algorithm=1, trees=3))
idx, dist = flann.knnSearch(des2, 2)
print(idx)
print(dist)
如果你的 cv2.flann_Index 搜索结果莫名其妙,可以检查一下是不是因为建立索引之后,把原始数据删除了