Windows中Python與OpenCV C++之間Mat傳遞

1 VS2017中創建DLL項目

1.1 創建DLL項目

1.2 刪除自動創建的文件

將stdafx.h、targetver.h、dllmain.cpp、MyDLL.cpp、stdafxc.pp刪除。

刪除後，記得要在C/C++》預編譯頭 中取消使用預編譯頭

1.3 配置OpenCV環境

這一步可參靠我的另一篇文章《OpenCV 3.2.0 + opencv_contrib+VS2017》。需要修改地方如下。

2 編寫調用OpenCV的C代碼

創建MyDLL.cpp文件，輸入如下代碼：

 
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/opencv.hpp>                   
#include   <stdlib.h>   
#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文件

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後，顯示如下。