，在蟒蛇

Question

OpenCV的KNN搜索我在C++中的OpenCV的此示例代码：

flann::KDTreeIndexParams indexParams; 
flann::Index kdtree(Mat(cloud2d).reshape(1), indexParams); 
vector<float> query; 
query.push_back(370); 
query.push_back(464); 
vector<int> indices; 
vector<float> dists; 
kdtree.knnSearch(query, indices, dists, 3); 

我该怎么做相同的蟒蛇？试过了，但不能用cv2创建kdtree或KDTreeIndexParams。

Answer 1

FLANN是一个ANN库，用C++编写，独立于OpenCV。它提供了Python的绑定，在pyflann。

一个例子可以发现here：

from pyflann import * 
import numpy as np 

dataset = np.array(
    [[1., 1, 1, 2, 3], 
    [10, 10, 10, 3, 2], 
    [100, 100, 2, 30, 1] 
    ]) 
testset = np.array(
    [[1., 1, 1, 1, 1], 
    [90, 90, 10, 10, 1] 
    ]) 
flann = FLANN() 
result, dists = flann.nn(
    dataset, testset, 2, algorithm="kmeans", branching=32, iterations=7, checks=16) 
print result 
print dists 

dataset = np.random.rand(10000, 128) 
testset = np.random.rand(1000, 128) 
flann = FLANN() 
result, dists = flann.nn(
    dataset, testset, 5, algorithm="kmeans", branching=32, iterations=7, checks=16) 
print result 
print dists 

这个例子应该足以让你开始。