python - 将基于 OpenCV 的 C++ 函数与 Mat/Numpy 转换为 Python 公开

Question

我在尝试使用 Numpy ndarray 将链接到 OpenCV 并使用 OpenCV 的 Mat 数据类型的动态 C++ 库函数公开给 Python 2.7 时遇到了问题.

我想出了一个类似于 lightalchemist's solution here 的解决方案我也尝试过使用 boost::python 和 boost::numpy(也链接到 Python 2.7)described in this SO question.

现在我坚持使用前者。我已经到了可以在 iPython 中加载模块的地步，并且我看到了一个我正在尝试使用 inspect 模块移植的函数，我什至可以调用它甚至执行它。但是，当我尝试使用 NumpyAllocator ( see lightalchemist's solution ) 类将 Mat 对象转换回 ndarray 时，问题就会出现。首先，当我尝试从外部 C++ 可执行文件调用 pyopencv_from 函数时，它在编码时使用 NumpyAllocator，它在

PyEnsureGIL gil;

，每次都没有消息。 Lightalchemist 的解决方案没有在 pyopencv_to 中使用它(编辑:如果传入的 ndarray 已经分配)，它似乎可以工作。但是，官方 OpenCV cv2.cpp 也确实在其中使用了分配器，因此如果我尝试使用该函数甚至无法将输入 ndarray 转换为 Mat。

当我尝试使用 iPython 中的模块时，它看到了函数。同样，它会正确执行它(将进度打印到控制台)，但是当它到达 pyopencv_from 时，它会出现段错误并终止 iPython shell。

编辑:我使用与 lightalchemist 完全相同的来源，除了我公开一个函数，与官方 OpenCV 端口相同:

static PyMethodDef methods[] = {
{"findEdgesCGTG", (PyCFunction)pycvex_findEdgesCGTG, METH_KEYWORDS, "findEdgesCGTG(source) -> edgeGradient, edgeOrientations"},
    {NULL, NULL}
};
extern "C"
#if defined WIN32 || defined _WIN32
__declspec(dllexport)
#endif
void initcvex()
{
  import_array();
  PyObject* m = Py_InitModule(MODULESTR, methods);
  PyObject* d = PyModule_GetDict(m);
  opencv_error = PyErr_NewException((char*)MODULESTR".error", NULL, NULL);
  PyDict_SetItemString(d, "error", opencv_error);
}

有人知道如何解决这个转换问题吗？

Answer 1

问题不是

PyEnsureGIL gil;

声明。只有在我尝试使用外部 C++ 函数时才行不通。当我在这里尝试将两个输出值打包成一个时，问题实际上发生了:

return Py_BuildValue("(NN)", pyopencv_from(edgeGrad), pyopencv_from(edgeOri));

临时解决方法是使用以下语句:

return pyopencv_from(edgeGrad);

编辑:对不起，我又搞砸了。 Py_BuildValue 工作得很好。刚刚再次测试了我的功能，它运行良好。一定是我上次测试的时候忘记放return关键字了，所以报了“error return without exception set”。

这是完整的功能代码(我希望它对任何将 OpenCV 的东西移植到 python 的人有用)

static PyObject* pycvex_findEdgesCGTG(PyObject* , PyObject* args, PyObject* kw)
{
    PyObject* pyobj_source = NULL;
    Mat source;
    Mat edgeGrad;
    Mat edgeOri;
    const char* keywords[] = { "src", NULL };
    if( PyArg_ParseTupleAndKeywords(args, kw, "O:findEdgesCGTG", (char**)keywords, &pyobj_source) &&
        pyopencv_to(pyobj_source, source));
    {
        ERRWRAP2(findEdgesCGTG(source,edgeGrad,edgeOri));
        return Py_BuildValue("(NN)", pyopencv_from(edgeGrad), pyopencv_from(edgeOri));
    }
    return NULL;
}

更新:

这是 github repo使用我为使用 OpenCV 的 Mat 类公开代码而编写的开放 C++ 代码，尽可能减少痛苦。这适用于 OpenCV 3.X。

对于 OpenCV 2.X，您可以使用 this code/Yati Sagade 的例子。如果您想使用 opencv Mat 类公开函数而不用担心显式转换，可以很容易地将我的代码中的 Booost 转换器与 Yati 代码中的转换函数相适应。