将 OpenCv Mat 从 C++ 传递给 Python

Passing OpenCv Mat from C++ to Python

本文关键字:Python C++ OpenCv Mat      更新时间:2023-10-16

我需要将OpenCv图像从C++发送到Python以对其进行一些处理。 垫子将通过代码接收,但为了简单起见,我在这里使用 imread 来回答这个问题。

我在代码的C++部分所做的是:

#include <Python.h>
#include <arrayobject.h>
#include <iostream>
#include <opencv2/opencv.hpp>

#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION

using namespace cv;
using namespace std;
int main(int argc, char *argv[])
{
QCoreApplication a(argc, argv);
Mat image = imread("test.jpg");
Py_Initialize();
PyObject *pName, *pModule, *pDict, *pFunc, *pArgs, *pValue;
pName = PyUnicode_FromString("prog");
if (pName == NULL)
{
PyErr_Print();
return 0;
}
pModule = PyImport_Import(pName);
if (pModule == NULL)
{
PyErr_Print();
return 0;
}
pDict = PyModule_GetDict(pModule);
pFunc = PyDict_GetItemString(pDict, "add");
if (pFunc == NULL)
{
PyErr_Print();
return 0;
}
pArgs = PyTuple_New(1);
import_array ();
npy_intp dimensions[3] = {image.rows, image.cols, image.channels()};
pValue = PyArray_SimpleNewFromData(image.dims + 1, (npy_intp*)&dimensions, NPY_UINT8, image.data);
PyTuple_SetItem(pArgs, 0, pValue);
PyObject* pResult = PyObject_CallObject(pFunc, pArgs);
if(pResult == NULL)
cout<<"Calling the add method failed"<<endl;
long result = PyLong_AsLong(pResult);
cout<<"Result = "<<result<<endl;
Py_Finalize();
return 0;
}

此代码编译并运行。

对于 Python 部分:

import cv2
import numpy as np
def add (a):
print ("Contents of a :")
print (a)
# mat_array = cv2.fromarray(a, numpy.float32)
vis0 = cv.fromarray(a)
return 0

Python 代码从 C++ 接收 numpy 数组(我认为(,当我打印a的内容时,我有一个输出(所以我认为我正在从C++接收图像(。

现在我需要将a中的数据转换为 Python 中的 cv2 Mat,以便我可以处理它。

一旦我到达mat_array = cv2.fromarray(a, numpy.float32)行或vis0 = cv.fromarray(a)代码崩溃并显示以下输出:

Exception ignored in: <module 'threading' from '/usr/lib/python3.5/threading.py'>
Traceback (most recent call last):
File "/usr/lib/python3.5/threading.py", line 1283, in _shutdown
assert tlock.locked()
SystemError: <built-in method locked of _thread.lock object at 0x7ff0f34d20d0> returned a result with an error set

如何正确发送/接收 Mat 对象?

请在这里找到我的答案。您也可以在此处找到其他答案。用于转换numpy -> cv::Matcv::Mat -> numpy.