python与C语言之间的数组传递

参考文章:https://mp.weixin.qq.com/s/f1nzrqV_qb6QbDct9wJ3sw

在Ubuntu系统下测试python调用C++程序,完成数组传入、传出、图片传输。其实图片就是特定数据类型的数组。

test.cpp:


#include "opencv2/opencv.hpp"


#define DLLEXPORT extern "C"
DLLEXPORT void sum_f(float* a,float* b,float* res,int n)
{
    for(int i=0; i<n; i++)
    {
        res[i] = a[i]+b[i];
    }
}

#define DLLEXPORT extern "C"
DLLEXPORT void test_opencv(unsigned char* img,int n)
{
    cv::Mat I = cv::Mat(n,n,CV_8UC3,img);
    cv::imwrite("ttt.jpg",I);
}

动态库编译

       g++ test.cpp -fpic -shared -o test.so `pkg-config --cflags --libs opencv`

       单引号为调用默认安装的opencv

test.py

from ctypes import cdll
import numpy as np
import ctypes

#导入动态库
mydll = cdll.LoadLibrary('test.so')

# 函数声明
sum_f = mydll.sum_f
sum_f.argtypes = [np.ctypeslib.ndpointer(dtype=np.float32,ndim=1,flags="C_CONTIGUOUS"),
                  np.ctypeslib.ndpointer(dtype=np.float32,ndim=1,flags="C_CONTIGUOUS"),
                  np.ctypeslib.ndpointer(dtype=np.float32,ndim=1,flags="C_CONTIGUOUS"),
                  ctypes.c_int]

x = np.ones(12,dtype=np.float32)*2
y = np.ones(12,dtype=np.float32)*3
print('x=',x)
print('y=',y)
res = np.zeros(12,dtype=np.float32)
#函数调用
sum_f(x,y,res,12)

print('x+y=',res)

# test opencv ===================================
test_opencv = mydll.test_opencv
test_opencv.argtypes = [np.ctypeslib.ndpointer(dtype=np.uint8,ndim=1,flags="C_CONTIGUOUS"),
                       ctypes.c_int]

n = 255
img = np.zeros((n,n,3),np.uint8)
img[:,:,0] = 255
img = img.reshape(-1)

test_opencv(img,n)

输入为x、y,输出为res,为两个之和。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值