从C ++向Python传递OpenCv Mat

时间:2018-08-30 06:51:08

标签: python c++ opencv

我需要将C ++的OpenCv映像发送到Python,以对其进行一些处理。 Mat将通过代码接收,但为简单起见,我在此处使用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对象?

1 个答案:

答案 0 :(得分:0)

请找到我的答案file_data_name。您还可以找到其他答案here。用于转换numpy -> cv::Matcv::Mat -> numpy