ctypes实现python和c之间的数据交互
2024-01-08 13:37:50
1、目的
将c/c++编译成动态链接库,通过python调用,实现常见变量比如int,字符串之间的交互,以及np.ndarray和cv::Mat间的交互.
2 步骤
新建CMakeLists.txt,如下,需要安装opencv,安装教程参考Ubuntu 18.04 安装opencv4.2.0,如果遇到IPPICV问题参考解决编译opencv时,卡在IPPICV
cmake_minimum_required(VERSION 3.10)
project(test)
add_definitions(-std=c++11)
add_definitions(-DAPI_EXPORTS)
set(CMAKE_CXX_STANDARD 11)
set(CMAKE_BUILD_TYPE Debug)
find_package(OpenCV)
include_directories(${OpenCV_INCLUDE_DIRS})
add_library(myplugins SHARED ${PROJECT_SOURCE_DIR}/test.cpp)
target_link_libraries(myplugins ${OpenCV_LIBS})
新建test.cpp
#include <opencv2/opencv.hpp>
#include <string>
using namespace std;
#if defined(_MSC_VER)
#define API __declspec(dllimport)
#else
#define API
#endif // API_EXPORTS
typedef struct test
{
/* data */
int ages;
std::string name;
cv::Mat pic;
} Person;
extern "C" API void * init(int age_, char * name_ptr, unsigned char *src_data, int rows, int cols)
{
Person *handle = new Person();
std::string name_ = name_ptr;
handle->ages = age_;
handle->name = name_;
handle->pic = cv::Mat(rows, cols, CV_8UC3, src_data);
return (void*) handle;
}
extern "C" API int get_ages(void* handle_)
{
Person * handle = (Person*)handle_;
return handle->ages;
}
extern "C" API char* get_name(void* handle_)
{
Person * handle = (Person*)handle_;
return (char*)handle->name.c_str();
}
extern "C" API void get_pic(void* handle_, int rows, int cols, unsigned char* out_data)
{
Person * handle = (Person*)handle_;
cv::imshow("raw", handle->pic);
cv::waitKey();
//数据处理
//....
//返回结果
memcpy(out_data, handle->pic.data, rows*cols*3);
}
新建test.py,具体转换见代码注释
from ctypes import *
import cv2
import numpy as np
import numpy.ctypeslib as npct
#加载链接库
lib = CDLL('./build/libmyplugins.so')
#定义函数的输入类型和输出类型
lib.get_ages.restype = c_int
lib.get_ages.argtypes = [c_void_p] #输入句柄(指针)
lib.get_name.restype = c_void_p #返回char*
lib.get_name.argtypes = [c_void_p] #输入句柄(指针)
img = cv2.imread("oil-bin-1.png") #修改成自己的图片
ages = 23
name = bytes("kitty", "utf-8") #要对应c函数中的char*,需要转为二进制
(rows, cols) = (img.shape[0], img.shape[1])
image_type = npct.ndpointer(dtype = np.uint8, ndim = 3, shape = img.shape, flags="C_CONTIGUOUS") #指定numpy图片的格式
lib.init.restype = c_void_p #返回句柄(指针)
lib.init.argtypes = [c_int, c_void_p, image_type, c_int, c_int]
#将np格式的图片传递到c
hd = lib.init(c_int(ages), c_char_p(name), img, c_int(rows), c_int(cols))
print(lib.get_ages(hd)) #返回整数,直接打印
name = c_char_p(lib.get_name(hd)).value #返回char*, 需要转换
print(name.decode('utf-8')) #解码成字符串
##第一种读取图片的方式,使用ndpointer指定图片类型
out_image = np.zeros_like(img).astype(np.uint8) #需要先开辟内存
lib.get_pic.argtypes = [c_void_p, c_int, c_int, image_type]
lib.get_pic(hd, c_int(rows), c_int(cols), out_image)
cv2.imshow("out1", out_image)
##第二种读取图片的方式,不使用ndpointer指定图片类型
out_image = np.zeros_like(img).astype(np.uint8) #需要先开辟内存
lib.get_pic.argtypes = [c_void_p, c_int, c_int, c_void_p] #句柄(指针),rows,cols, 图片指针
lib.get_pic(hd, c_int(rows), c_int(cols), out_image.ctypes.data_as(POINTER(c_ubyte)))
cv2.imshow("out2", out_image)
cv2.waitKey()
3、测试
将上述3个文件放到同一个目录,然后在命令行中进入到该目录,依次运行,会弹出图片,单击图片,按任意键即可
mkdir build
cd build
cmake ..
make
cd ..
python test.py
文章来源:https://blog.csdn.net/mjlsuccess/article/details/135454167
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!