最近看到一個巨牛的人工智能教程,分享一下給大家。教程不僅是零基礎,通俗易懂,而且非常風趣幽默,像看小說一樣!覺得太牛了,所以分享給大家。平時碎片時間可以當小說看,【點這裏可以去膜拜一下大神的“小說”】。
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後,顯示如下。
6 源碼
https://github.com/huachao1001/blog/tree/master/python_cpp_opencv