c++ python opencv_Windows中Python与OpenCV C++之间Mat传递

最近看到一个巨牛的人工智能教程，分享一下给大家。教程不仅是零基础，通俗易懂，而且非常风趣幽默，像看小说一样！觉得太牛了，所以分享给大家。平时碎片时间可以当小说看，【点这里可以去膜拜一下大神的“小说”】。

1 VS2017中创建DLL项目

1.1 创建DLL项目

创建DLL项目

1.2 删除自动创建的文件

将stdafx.h、targetver.h、dllmain.cpp、MyDLL.cpp、stdafxc.pp删除。

删除自动创建的文件

删除后，记得要在C/C++》预编译头 中取消使用预编译头

取消使用预编译头

1.3 配置OpenCV环境

添加包含目录和库目录

添加附加依赖项

2 编写调用OpenCV的C代码

创建MyDLL.cpp文件，输入如下代码：

#include

#include

#include

#define DLLEXPORT extern "C" __declspec(dllexport)

using namespace cv;

DLLEXPORT uchar* cpp_canny(int height, int width, uchar* data) {

cv::Mat src(height, width, CV_8UC1, data);

cv::Mat dst;

Canny(src, dst, 100, 200);

uchar* buffer = (uchar*)malloc(sizeof(uchar)*height*width);

memcpy(buffer, dst.data, height*width);

return buffer;

}

DLLEXPORT void release(uchar* data) {

free(data);

}

上代码简单调用了OpenCV的Canny函数，使用指针传递数据。值得注意的是，自定义的宏DLLEXPORT用于指定暴露的函数，只有加了这个宏的函数在外面才能调用。

注意： 由于cpp_canny函数中申请了内存空间，需要释放，因此加了函数release，用于在python端释放内存。

3 编译生成DLL

执行

编译生成DLL

在项目目录中可以找到生成的DLL文件

生成的DLL文件

4 编写Python代码

创建Python文件，输入如下代码：

import cv2

from numpy.ctypeslib import ndpointer

import ctypes

import numpy as np

dll=ctypes.WinDLL('MyDLL.dll')

def cpp_canny(input):

if len(img.shape)>=3 and img.shape[-1]>1:

gray=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)

h,w=gray.shape[0],gray.shape[1]

# 获取numpy对象的数据指针

frame_data = np.asarray(gray, dtype=np.uint8)

frame_data = frame_data.ctypes.data_as(ctypes.c_char_p)

# 设置输出数据类型为uint8的指针

dll.cpp_canny.restype = ctypes.POINTER(ctypes.c_uint8)

# 调用dll里的cpp_canny函数

pointer = dll.cpp_canny(h,w,frame_data)

# 从指针指向的地址中读取数据，并转为numpy array

np_canny = np.array(np.fromiter(pointer, dtype=np.uint8, count=h*w))

return pointer,np_canny.reshape((h,w))

img=cv2.imread('input.png')

ptr,canny=cpp_canny(img)

cv2.imshow('canny',canny)

cv2.waitKey(2000)

#将内存释放

dll.release(ptr)

5 执行

将VS2017生成的MyDLL.dll文件拷贝到python代码根目录，并加入测试图片input.jpg，示例如下。

相关文件

执行test.py后，显示如下。

输出结果

6 源码