C++使用Py*調(diào)用Python3模塊中類成員函數(shù)及數(shù)組參數(shù)傳遞

1.首先來看Python模塊的部分結(jié)構(gòu)和代碼麸折。ssd_network_classify.py文件中有SSD_Network_Classify類及其識別的成員函數(shù)detect_image()行疏,返回值是一個1維的不定長double型數(shù)組。

class SSD_Network_Classify:
    
     #其他函數(shù)實現(xiàn)省略雄右。。掉房。
     
        def detect_image(self, img_raw=None):
        '''
        獲取圖片數(shù)據(jù)進行檢測仗岖。
        :param img_raw: 從c++傳來的一維480*640*3大小的int型圖片數(shù)據(jù)(包含負值)
        :return: 返回的是一維數(shù)組
        '''
        #將原始圖片數(shù)據(jù)shape成480*640*3的uint8類型數(shù)據(jù)。
        img_data = np.reshape(img_raw, (480, 640, 3)).astype(np.uint8)
        #以下兩行輸出用于顯示傳參的值幫助理解擎析。
        print(img_data.shape)
        print(img_data)
        #進行目標檢測并返回檢測結(jié)果簿盅。
        rclasses, rscores, rbboxes = self.detect(img_data)
        #顯示檢測結(jié)果
        cv2.imshow('DetectImage', img_data)
        cv2.waitKey(1)
        #以下將返回的結(jié)果拼接成一維的數(shù)組返回給調(diào)用該函數(shù)的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中輸出傳入的參數(shù)(從c++傳送過來)示例如下:


data.png

2. C++端獲取攝像頭數(shù)據(jù)的類CameraBase:

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

//以下是.cpp中主要成員函數(shù)的實現(xiàn)
"構(gòu)造函數(shù)揍魂,打開攝像頭"
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";
    }
}

"讀取網(wǎng)絡攝像頭的圖片數(shù)據(jù)"
void CameraBase::grabImages() {
    this->capture->read(this->bgr_image);
}

"圖片的Mat數(shù)據(jù)轉(zhuǎn)換成char類型的數(shù)組數(shù)據(jù)"
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++調(diào)用Python的函數(shù)callPythonForDetect。這塊代碼是c++調(diào)用Python的核心代碼现斋,具體代碼解釋已經(jīng)在代碼注釋中寫得很清楚了喜最。

int callPythonForDetect() {
    //調(diào)用web攝像頭進行處理
    CameraBase *camera = new CameraBase();
    
    
    //------------------以下是調(diào)用Python模塊的代碼------------------//
    //python環(huán)境初始化
    Py_Initialize();
    if (!Py_IsInitialized())
        return -1;
    
    //導入系統(tǒng)包用于擴展需要加載的Python模塊的路徑,否則即使Python模塊在當前目錄也無法加載
    PyRun_SimpleString("import sys \nsys.argv = ['']");
    
    //加載Python模塊的路徑
    PyRun_SimpleString("sys.path.append('/absolute/path/to/python/module')");
   
    //導入需要調(diào)用的模塊
    PyObject *pyModule = PyImport_ImportModule("ssd_network_classify");
    if (!pyModule) {
        printf("Can not open python module\n");
        return -1;
    }
    
    
    //獲取python模塊中的類名并創(chuàng)建對象實例
    PyObject *pyClass = PyObject_GetAttrString(pyModule, "SSD_Network_Classify");
    PyObject *pyClassInstance = PyObject_CallObject(pyClass, NULL);
    
    //獲取Python模塊中相應的函數(shù)名
    PyObject *pyFunc = PyObject_GetAttrString(pyClass, "detect_image");

    //聲明或定義變量
    npy_intp IMGSHAPE[1] = {480 * 640 * 3};//圖片數(shù)據(jù)的shape參數(shù)值
    char *img_data = new char[IMGSHAPE[0]];//從攝像頭中獲取的圖片數(shù)據(jù)保存的變量
    PyByteArrayObject *pyIMgArr;//image數(shù)組的Python對象
    
    //設置發(fā)送給Python函數(shù)的參數(shù)對象
    PyObject *pyArgs = PyTuple_New(2); 
    while (true) {
        camera->grabImages();//獲取攝像頭數(shù)據(jù)
        
        //獲取image數(shù)據(jù)并保存至img_data數(shù)組中
        camera->bgrImageMatToArray(img_data);
        
        //必須添加如下函數(shù)庄蹋,否則無法執(zhí)行PyArray_SimpleNewFromData
        import_array ();
        
        //將c的img數(shù)組數(shù)據(jù)轉(zhuǎn)換成pyobject類型的數(shù)組數(shù)據(jù)
        pyIMgArr = reinterpret_cast<PyByteArrayObject *>
        (PyArray_SimpleNewFromData(1, IMGSHAPE, NPY_BYTE, reinterpret_cast<void *>(img_data)));
        
        //設置調(diào)用函數(shù)的self值為前面該類創(chuàng)建的實例瞬内,否則無法使用self變量進行調(diào)用而出錯
        PyTuple_SetItem(pyArgs, 0, Py_BuildValue("O", pyClassInstance));
       
        //設置變量的第二個參數(shù)值為byte類型的數(shù)組作為圖片數(shù)據(jù)
        PyTuple_SetItem(pyArgs, 1, reinterpret_cast<PyObject *>(pyIMgArr));
        
        //調(diào)用python函數(shù)進行識別任務并返回相應的結(jié)果
        PyObject *pyResult = PyObject_CallObject(pyFunc, pyArgs);
        
        
        //以下是對返回的一維數(shù)組結(jié)果進行處理
        if (pyResult) {
            //將結(jié)果類型轉(zhuǎn)換成數(shù)組對象類型
            PyArrayObject *pyResultArr = (PyArrayObject *) pyResult;
            
            //也可以使用以下兩行代碼來代替上面的類型轉(zhuǎn)換。
            //PyArray_Descr *descr = PyArray_DescrFromType(NPY_DOUBLE);
            //PyArrayObject *pyResultArr = (PyArrayObject*)PyArray_FromAny(pyResult, descr,1,1,NPY_ARRAY_C_CONTIGUOUS,NULL);
            
            //從Python中的PyArrayObject解析出數(shù)組數(shù)據(jù)為c的double類型限书。
            double *resDataArr = (double *) PyArray_DATA(pyResultArr);
            int dimNum = PyArray_NDIM(pyResultArr);//返回數(shù)組的維度數(shù)虫蝶,此處恒為1
            npy_intp *pdim = PyArray_DIMS(pyResultArr);//返回數(shù)組各維度上的元素個數(shù)值
            
            
            //以下是對返回結(jié)果的輸出顯示
            for (int i = 0; i < dimNum; ++i) {
                for (int j = 0; j < pdim[0]; ++j)
                    cout << resDataArr[i * pdim[0] + j] << ",";
            }
            cout << endl;
        }
    }
    //釋放Python環(huán)境
    Py_Finalize();
}

從調(diào)用的Python函數(shù)中返回的數(shù)組結(jié)果示例如下:


dataout.png

以下是最終C++通過調(diào)用Python版本實現(xiàn)的目標識別網(wǎng)絡展示的結(jié)果:


result.png

總結(jié):
*** 使用C++調(diào)用python3模塊接口的示例基本沒有,有的大部分都是python2版本的示例蔗包,而新版本的很多函數(shù)名稱和用法改變都很大秉扑,導致我在寫這塊代碼的時候碰到很多問題,就這個簡單需求花了我整整3天的時間调限,期間有考慮使用第三方框架進行解決,但是發(fā)現(xiàn)也很麻煩误澳。其中需要注意的是耻矮,因為Python的類函數(shù)的第一個參數(shù)是self,并且是傳入對象本身忆谓,因此在c中調(diào)用的時候也要考慮為其賦值(PyTuple_SetItem(pyArgs, 0, Py_BuildValue("O", pyClassInstance)); 這行代碼很重要)裆装,否則會出錯。很多例子都考慮的是調(diào)用非類成員函數(shù),因此不需要考慮self變量而比較容易哨免。我是自己嘗試很久并經(jīng)過調(diào)試才知道該怎么給self賦值的茎活,這也是我寫這篇文章的原因所在,希望給其他需要的人一些參考琢唾,少走彎路载荔。***

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市采桃,隨后出現(xiàn)的幾起案子懒熙,更是在濱河造成了極大的恐慌,老刑警劉巖普办,帶你破解...
    沈念sama閱讀 212,542評論 6 493
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件工扎,死亡現(xiàn)場離奇詭異,居然都是意外死亡衔蹲,警方通過查閱死者的電腦和手機肢娘,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,596評論 3 385
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來舆驶,“玉大人橱健,你說我怎么就攤上這事≌暝叮” “怎么了畴博?”我有些...
    開封第一講書人閱讀 158,021評論 0 348
  • 文/不壞的土叔 我叫張陵,是天一觀的道長蓝仲。 經(jīng)常有香客問我俱病,道長,這世上最難降的妖魔是什么袱结? 我笑而不...
    開封第一講書人閱讀 56,682評論 1 284
  • 正文 為了忘掉前任亮隙,我火速辦了婚禮,結(jié)果婚禮上垢夹,老公的妹妹穿的比我還像新娘溢吻。我一直安慰自己,他們只是感情好果元,可當我...
    茶點故事閱讀 65,792評論 6 386
  • 文/花漫 我一把揭開白布促王。 她就那樣靜靜地躺著,像睡著了一般而晒。 火紅的嫁衣襯著肌膚如雪蝇狼。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 49,985評論 1 291
  • 那天倡怎,我揣著相機與錄音迅耘,去河邊找鬼贱枣。 笑死,一個胖子當著我的面吹牛颤专,可吹牛的內(nèi)容都是我干的纽哥。 我是一名探鬼主播,決...
    沈念sama閱讀 39,107評論 3 410
  • 文/蒼蘭香墨 我猛地睜開眼栖秕,長吁一口氣:“原來是場噩夢啊……” “哼春塌!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起累魔,我...
    開封第一講書人閱讀 37,845評論 0 268
  • 序言:老撾萬榮一對情侶失蹤摔笤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后垦写,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體吕世,經(jīng)...
    沈念sama閱讀 44,299評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 36,612評論 2 327
  • 正文 我和宋清朗相戀三年梯投,在試婚紗的時候發(fā)現(xiàn)自己被綠了命辖。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 38,747評論 1 341
  • 序言:一個原本活蹦亂跳的男人離奇死亡分蓖,死狀恐怖尔艇,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情么鹤,我是刑警寧澤终娃,帶...
    沈念sama閱讀 34,441評論 4 333
  • 正文 年R本政府宣布,位于F島的核電站蒸甜,受9級特大地震影響棠耕,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜柠新,卻給世界環(huán)境...
    茶點故事閱讀 40,072評論 3 317
  • 文/蒙蒙 一窍荧、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧恨憎,春花似錦蕊退、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,828評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至钥组,卻和暖如春茉贡,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背者铜。 一陣腳步聲響...
    開封第一講書人閱讀 32,069評論 1 267
  • 我被黑心中介騙來泰國打工腔丧, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人作烟。 一個月前我還...
    沈念sama閱讀 46,545評論 2 362
  • 正文 我出身青樓愉粤,卻偏偏與公主長得像,于是被迫代替她去往敵國和親拿撩。 傳聞我的和親對象是個殘疾皇子衣厘,可洞房花燭夜當晚...
    茶點故事閱讀 43,658評論 2 350

推薦閱讀更多精彩內(nèi)容