从python中保存opencv FlannBasedMatcher

Question

我正在尝试保存创建的FlannBasedMatcher，并使用OpenCV-python4.5.5.62进行培训。

类的save函数只保存参数。我也可以访问trainDescriptors，但是我想要的是拯救曾经受过训练的flann。能够从另一个python脚本重新加载它。

有人知道这件事吗？

代码示例可能类似于(我知道代码是无用和丑陋的，但是一个示例有时可以帮助理解问题)：

import cv2
import numpy as np

# Creating random data
datas = np.random.rand(50000, 512)
datas = np.float32(datas)

# Creating the FlannBasedMatcher and train it
flann = cv2.FlannBasedMatcher(index_params, search_params)
flann.add([datas])
flann.train()

# Here I am able to do knnMatch search in 'flann'

# Not existing function that I am looking for
flann.saveTrainedModel("my_file_path")

# Loading from an existing file 
flannLoaded = cv2.FlannBasedMatcher.LoadFromFile("my_file_path")

# Here I want to be able to do knnMatch in 'flannLoaded' without training it.

如果有人对像saveTrainedModel和/或LoadFromFile这样的函数有线索，这会有很大的帮助。

编辑:我在C++：https://github.com/renweizhukov/LearningOpenCV/blob/master/FlannKnnSavableMatching1toN/src/FlannBasedSavableMatcher.cpp中找到了一些可以在网上做这件事，但是它使用的是在python版本中不可用的trainDescCollection。

非常感谢你的帮助，

Answer 1

我终于找到了一种通过使用pyflann模块来实现它的方法。首先，我必须修改pyflann (pyflann-1.6.14)和索引 (index-0.2-py37)模块才能接受python 3的使用。

然后您可以保存flann索引：

from pyflann.index import FLANN, set_distance_type
 
# creating the data
datas = np.random.rand(1000000, 512)
datas = np.float32(datas)
 
# declaring the FLANN
flann = FLANN()
# build the index from datas
flann.build_index(datas, algorithm="linear", checks=10)
# Saving the index created therebefore
flann.save_index(b"my_saved_flann.txt")
 
# Searched data
searched = datas[i]
searched = np.expand_dims(searched, axis=0)
 
# retrieving the 500 nearest neighbours of searched
results, distances = flann.nn_index(searched, 500)