PyFlann 安装方法
pip install pyflann
按照以下更改代码:
https://github.com/primetang/pyflann/issues/1
然后可运行以下代码:
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)
接下来就根据自己的实际问题进行FLANN 之旅了。。
使用方法参考:http://blog.csdn.net/jcq521045349/article/details/78898971