OpenCV 将数据从 C++ 传递到 Python

Question

使用opencv中的 opencv 时，我在尝试使 SIFT 工作时遇到了麻烦（ 这是问题所在），所以我在 C++ 中实现了我需要的东西：

  cv::Mat src = cv::imread(argv[1], cv::IMREAD_GRAYSCALE);
  auto sift = cv::xfeatures2d::SIFT::create();
  vector<cv::KeyPoint> keypoint;
  sift->detect(src, keypoint);
  cv::Mat descriptor;
  sift->compute(src, keypoint, descriptor);

我需要的是KeyPoint的向量和描述符矩阵。

现在我想继续使用 Python 因为我需要 numpy 并且我需要一种传输向量的方法。

有没有办法将向量移动到 Python？

对于矩阵，我可以将其写入文件，然后从 python 读取，但我不知道如何处理 KeyPoints 向量。

我在想我可以将代码移动到 function 中，将其编译为共享库并尝试使用 cffi 之类的东西导入它，但对于 C++，问题是我对此几乎没有经验，我认为会有问题类型的处理。

Answer 1

您应该枚举一个std::vector并转换每个 C++ KeyPoint object。 由于要将关键点转换为矩阵，请这样做。

std::vector<cv::KeyPoint> points;
...
cv::Mat_<float> mat(points.size(), 7);
for (int i = 0; i < mat.rows; i++) {
    auto& p = points[i];
    mat.row(i) <<
        p.pt.x, p.pt.y, p.size, p.angle, p.response, p.octave, p.class_id;
}