OpenCV python的API：FlannBasedMatcher

Question

我正在尝试重写此处描述的代码。使用适用于Opencv的python API。

代码的第3步具有以下几行：

 FlannBasedMatcher matcher;
 std::vector< DMatch > matches;
 matcher.match( descriptors_object, descriptors_scene, matches );

我在OpenCV参考资料中反复查看，但是没有发现与python中的FlannBasedMatcher或其他可以完成工作的对象相关的内容。

有任何想法吗？

注意：我使用的是OpenCV 2.3.1和Python 2.6

Answer 1

在python2文件夹下的OpenCV
2.3.1提供的示例中，我发现了一个基于flann的match函数的实现，该函数不依赖于FlanBasedMatcher对象。

这是代码：

FLANN_INDEX_KDTREE = 1  # bug: flann enums are missing

flann_params = dict(algorithm = FLANN_INDEX_KDTREE,
                    trees = 4)

def match_flann(desc1, desc2, r_threshold = 0.6):
    flann = cv2.flann_Index(desc2, flann_params)
    idx2, dist = flann.knnSearch(desc1, 2, params = {}) # bug: need to provide empty dict
    mask = dist[:,0] / dist[:,1] < r_threshold
    idx1 = np.arange(len(desc1))
    pairs = np.int32( zip(idx1, idx2[:,0]) )
    return pairs[mask]