首页 > 编程语言 >C++使用Py*调用Python3模块中类成员函数及数组参数传递

C++使用Py*调用Python3模块中类成员函数及数组参数传递

时间:2023-08-11 15:35:16浏览次数:65  
标签:调用 img Python image Py C++ PyObject 数组 中类

1.首先来看Python模块的部分结构和代码。ssd_network_classify.py文件中有SSD_Network_Classify类及其识别的成员函数detect_image(),返回值是一个1维的不定长double型数组。

class SSD_Network_Classify:
    
     #其他函数实现省略。。。
     
        def detect_image(self, img_raw=None):
        '''
        获取图片数据进行检测。
        :param img_raw: 从c++传来的一维480*640*3大小的int型图片数据(包含负值)
        :return: 返回的是一维数组
        '''
        #将原始图片数据shape成480*640*3的uint8类型数据。
        img_data = np.reshape(img_raw, (480, 640, 3)).astype(np.uint8)
        #以下两行输出用于显示传参的值帮助理解。
        print(img_data.shape)
        print(img_data)
        #进行目标检测并返回检测结果。
        rclasses, rscores, rbboxes = self.detect(img_data)
        #显示检测结果
        cv2.imshow('DetectImage', img_data)
        cv2.waitKey(1)
        #以下将返回的结果拼接成一维的数组返回给调用该函数的c程序中。
        rclasses = np.reshape(rclasses, (-1, 1))
        rscores = np.reshape(rscores, (-1, 1))
        result = np.concatenate([rclasses, rscores, rbboxes], axis=1)
        result = np.squeeze(np.reshape(result, (1, -1)))
        #print(result)
        return result

在Python中输出传入的参数(从c++传送过来)示例如下:


 

2. C++端获取摄像头数据的类CameraBase:

//以下是.h文件中的成员变量
//cv::VideoCapture *capture;
//cv::Mat bgr_image;
//int device_num;

//以下是.cpp中主要成员函数的实现
"构造函数,打开摄像头"
CameraBase::CameraBase(int dev_num) {
    this->device_num = dev_num;
    this->capture = new cv::VideoCapture(this->device_num);
    if (!capture->isOpened()) {
        std::cout << "camera open failed\n";
    }
}

"读取网络摄像头的图片数据"
void CameraBase::grabImages() {
    this->capture->read(this->bgr_image);
}

"图片的Mat数据转换成char类型的数组数据"
char *CameraBase::bgrImageMatToArray( char *img_arr) {
    size_t img_size = this->bgr_image.total() * this->bgr_image.elemSize() ;
    std::memcpy(img_arr, this->bgr_image.data, img_size * sizeof(char));
    return img_arr;
}

3.核心部分。main.cpp中c++调用Python的函数callPythonForDetect。这块代码是c++调用Python的核心代码,具体代码解释已经在代码注释中写得很清楚了。

int callPythonForDetect() {
    //调用web摄像头进行处理
    CameraBase *camera = new CameraBase();
    
    
    //------------------以下是调用Python模块的代码------------------//
    //python环境初始化
    Py_Initialize();
    if (!Py_IsInitialized())
        return -1;
    
    //导入系统包用于扩展需要加载的Python模块的路径,否则即使Python模块在当前目录也无法加载
    PyRun_SimpleString("import sys \nsys.argv = ['']");
    
    //加载Python模块的路径
    PyRun_SimpleString("sys.path.append('/absolute/path/to/python/module')");
   
    //导入需要调用的模块
    PyObject *pyModule = PyImport_ImportModule("ssd_network_classify");
    if (!pyModule) {
        printf("Can not open python module\n");
        return -1;
    }
    
    
    //获取python模块中的类名并创建对象实例
    PyObject *pyClass = PyObject_GetAttrString(pyModule, "SSD_Network_Classify");
    PyObject *pyClassInstance = PyObject_CallObject(pyClass, NULL);
    
    //获取Python模块中相应的函数名
    PyObject *pyFunc = PyObject_GetAttrString(pyClass, "detect_image");

    //声明或定义变量
    npy_intp IMGSHAPE[1] = {480 * 640 * 3};//图片数据的shape参数值
    char *img_data = new char[IMGSHAPE[0]];//从摄像头中获取的图片数据保存的变量
    PyByteArrayObject *pyIMgArr;//image数组的Python对象
    
    //设置发送给Python函数的参数对象
    PyObject *pyArgs = PyTuple_New(2); 
    while (true) {
        camera->grabImages();//获取摄像头数据
        
        //获取image数据并保存至img_data数组中
        camera->bgrImageMatToArray(img_data);
        
        //必须添加如下函数,否则无法执行PyArray_SimpleNewFromData
        import_array ();
        
        //将c的img数组数据转换成pyobject类型的数组数据
        pyIMgArr = reinterpret_cast<PyByteArrayObject *>
        (PyArray_SimpleNewFromData(1, IMGSHAPE, NPY_BYTE, reinterpret_cast<void *>(img_data)));
        
        //设置调用函数的self值为前面该类创建的实例,否则无法使用self变量进行调用而出错
        PyTuple_SetItem(pyArgs, 0, Py_BuildValue("O", pyClassInstance));
       
        //设置变量的第二个参数值为byte类型的数组作为图片数据
        PyTuple_SetItem(pyArgs, 1, reinterpret_cast<PyObject *>(pyIMgArr));
        
        //调用python函数进行识别任务并返回相应的结果
        PyObject *pyResult = PyObject_CallObject(pyFunc, pyArgs);
        
        
        //以下是对返回的一维数组结果进行处理
        if (pyResult) {
            //将结果类型转换成数组对象类型
            PyArrayObject *pyResultArr = (PyArrayObject *) pyResult;
            
            //也可以使用以下两行代码来代替上面的类型转换。
            //PyArray_Descr *descr = PyArray_DescrFromType(NPY_DOUBLE);
            //PyArrayObject *pyResultArr = (PyArrayObject*)PyArray_FromAny(pyResult, descr,1,1,NPY_ARRAY_C_CONTIGUOUS,NULL);
            
            //从Python中的PyArrayObject解析出数组数据为c的double类型。
            double *resDataArr = (double *) PyArray_DATA(pyResultArr);
            int dimNum = PyArray_NDIM(pyResultArr);//返回数组的维度数,此处恒为1
            npy_intp *pdim = PyArray_DIMS(pyResultArr);//返回数组各维度上的元素个数值
            
            
            //以下是对返回结果的输出显示
            for (int i = 0; i < dimNum; ++i) {
                for (int j = 0; j < pdim[0]; ++j)
                    cout << resDataArr[i * pdim[0] + j] << ",";
            }
            cout << endl;
        }
    }
    //释放Python环境
    Py_Finalize();
}

从调用的Python函数中返回的数组结果示例如下:


 

以下是最终C++通过调用Python版本实现的目标识别网络展示的结果:


总结:
*** 使用C++调用python3模块接口的示例基本没有,有的大部分都是python2版本的示例,而新版本的很多函数名称和用法改变都很大,导致我在写这块代码的时候碰到很多问题,就这个简单需求花了我整整3天的时间,期间有考虑使用第三方框架进行解决,但是发现也很麻烦。其中需要注意的是,因为Python的类函数的第一个参数是self,并且是传入对象本身,因此在c中调用的时候也要考虑为其赋值(PyTuple_SetItem(pyArgs, 0, Py_BuildValue("O", pyClassInstance)); 这行代码很重要),否则会出错。很多例子都考虑的是调用非类成员函数,因此不需要考虑self变量而比较容易。我是自己尝试很久并经过调试才知道该怎么给self赋值的,这也是我写这篇文章的原因所在,希望给其他需要的人一些参考,少走弯路。***

标签:调用,img,Python,image,Py,C++,PyObject,数组,中类
From: https://www.cnblogs.com/lidabo/p/17623109.html

相关文章

  • c++ 使用移动语义来提高 vector 性能
    本文学习了微软的官方实例,用于理解std::move语义。#pragmaonce#include<iostream>#include<algorithm>#include<vector>usingnamespacestd;classMemoryBlock{public://Simpleconstructorthatinitializestheresource.explicitMemoryBl......
  • python限制函数的执行时间
    importsignalclassTimeoutException(Exception):passdeftimeout_handler(signum,frame):raiseTimeoutException()deflimit_time(seconds=10):defdecorator(func):defwrapper(*args,**kwargs):signal.signal(signal.SIGAL......
  • 某公司笔试题 - 计算字符串最后一个单词的长度(附python代码)
    题目:计算字符串最后一个单词的长度,单词以空格隔开,字符串长度小于5000(字符串末尾不变以空格未结尾)。python代码:str=input("请输入一个字符串:")iflen(str)<5000andlen(str)>0:#去掉字符串前后的空格s=str.strip("")#已空格为标记,对字符串切片......
  • C++调用Python传入参数、图片并接受返回值
    最近在做C++调用Pytorch模型进行识别的任务,C++关于Pytorch的教程很少,基本上都是用Python写的,但因为要识别任务是实时的,Python的执行效率不如C++,所以主题代码还是没用Python。网上利用C++调用Pytorch模型的方法主要是把模型文件转化成C++可以加载和执行的模型文件,利用的是TorchS......
  • python中偏函数--parital
    以往我没写一个函数并调用def_xx(a1,a2):returna1+a2data=_xx(11,22)print(data)使用partial函数之后fromfunctoolsimportpartialdef_xx(a1,a2):returna1+a2yy=partial(_xx,a2=100)#偏函数,包裹一个函数,并可以默认传一个参数data=y......
  • ​python爬虫——爬虫伪装和反“反爬”
    前言爬虫伪装和反“反爬”是在爬虫领域中非常重要的话题。伪装可以让你的爬虫看起来更像普通的浏览器或者应用程序,从而减少被服务器封禁的风险;反“反爬”则是应对服务器加强的反爬虫机制。下面将详细介绍一些常见的伪装和反反爬技巧,并提供对应的代码案例。1.User-Agent伪装User......
  • python调用zabbix api接口实时展示数据
    zabbixapi接口来进行展示。经过思考之后,计划获取如下内容:  1、  获得认证密钥  2、  获取zabbix所有的主机组  3、  获取单个组下的所有主机  4、  获取某个主机下的所有监控项  5、  获取某个监控项的历史数据  6、  获取某个监控项的......
  • C++ #pragma once指令:保护C++头文件不被重复包含
    一、#ifndef/#define/#endif指令的问题在C++中,头文件的作用就是将代码以模块的形式组织起来,便于复用和维护。但是,头文件很容易出现重复定义的问题。比如,某个头文件被多个源文件包含,这些源文件又有可能被其他源文件包含,那么就有可能出现一个头文件被重复包含的情况。这样就会......
  • C++多线程不加锁操作同一个整数
    #include<iostream>#include<thread>#include<vector>#include<chrono>#include<atomic>usingnamespacestd;intnum=0;//volatileintnum=0;//atomic<int>num=0;voidadd(){inttimes=0;for(int......
  • 《最新出炉》系列初窥篇-Python+Playwright自动化测试-11-playwright操作iframe-上篇
    1.简介原估计宏哥这里就不对iframe这个知识点做介绍和讲解了,因为前边的窗口切换就为这种网页处理提供了思路,另一个原因就是虽然iframe很强大,但是现在很少有网站用它了。但是还是有小伙伴或者童鞋们私下问这个问题,那么宏哥就单独写一篇关于iframe网页处理的文章。iframe是web自动......