OpenCV Flann

最新推荐文章于 2025-04-20 16:39:33 发布

mysticalwing

最新推荐文章于 2025-04-20 16:39:33 发布

阅读量1.1k

点赞数

分类专栏： OpenCV

本文链接：https://blog.csdn.net/Anderson_Y/article/details/106541685

版权

OpenCV Flann

文章目录

OpenCV Flann

简介

本文主要记录如何在python中使用opencv的flann模块进行图像点的最近邻搜索（仅作为作者自己的记录）。

FLANN 是一个用于高维空间数据的近似最近邻搜索的库，是一个经常会被用到的库。OpenCV flann 也实现了flann，其用途主要是用在特征匹配上面。这一部分可以参考 OpenCV Python tutorial Feature Matching 。下面介绍如何使用opencv-flann进行点的最近邻搜索。

使用 opencv flann_index 进行最近邻点搜索

这个是后面查资料时，在so上看到的。发现相比于 cv2.FlannBasedMatcher ，cv2.flann_Index 用于最近邻搜索要简单得多，也更合适。因此做一个补充，并移到前面来。

代码如下：

import cv2
import numpy as np
from matplotlib import pyplot as plt

if __name__ == '__main__':
    # generate dataset
    train_dataset_pts_nx2 = np.random.rand(1000, 2) * 1000

    # create KD-Tree object
    FLANN_INDEX_KDTREE = 1  # cv2 dost not contain FLANN_INDEX_KDTREE
    index_param = {
   "algorithm": FLANN_INDEX_KDTREE, "trees": 1}

    # generate dataset, the input must be float32
    flann_index = cv2.flann_Index(np.float32(train_dataset_pts_nx2), 
                                  index_param)

    # generate query points
    query_pts_nx2 = np.random.rand(100, 2) * 1000

    # query points must be float32 also
    indices, dists = flann_index.knnSearch(np.float32(query_pts_nx2), 
                                           knn=1)
    indices_nx1 = np.squeeze(