2017-02-12 6 views
4

PythonをC++アプリケーションに埋め込むことに関して質問があります。セットアップは以下の通りです:私はいくつかのデータを生成する(リアルタイムでイメージをレンダリングする)、それらを表示する大規模なC++アプリケーションを持っています。私はまた、これらの画像を受け入れるテンソルフローを使って、Pythonでニューラルネットワークを訓練しました。C++でPythonを埋め込む際のテンソルフローをインポートするとNULLが返される

私の考えは、pythonを埋め込み、numpy配列としてデータを送信し、ニューラルネットワークを使用して予測し、別の処理されたnumpy配列を戻して(C++で)表示することでした。私はPythonの上にテンソルフローを持たない基本テストをいくつか行って、PythonをCなどに埋め込む感覚を得ました。それはうまくいくようです。

しかし、いったん私がインポートしたいPythonスクリプトに "import tensorflow"を配置すると、PyImport_ImportModuleからC++パートにNULLが返されます。

import numpy as np 
def foo(img): 
    return np.clip(img * 2.0, 0, 255).astype(np.uint8) 

が問題ありません。しかし、次にはない。第2のケースで

import numpy as np 
import tensorflow as tf #this causes the fail 

def foo(img): 
    return np.clip(img * 2.0, 0, 255).astype(np.uint8) 

を、私はまだそれがCUDAなどを発見したが、その後、モジュールのインポートが失敗したことをtensorflowから標準出力にメッセージを取得します。

私の設定は、Windows 10 x64、Anaconda Python 3.5、tensorflow-0.12およびCUDA 8です。誰も同じような問題を抱えていましたか?私がテストした他のモジュール(numpy、pil、scipy)はうまく読み込めるようです。

解決できないように見える場合は、私はC++の部分とPythonの間で何らかの種類のIPCを利用します。

答えて

2

私はこの問題を解決しました。 PySys_SetArgvでargcとargvを設定する必要がありました。私はPyErr_Occurred()とPyErr_Print()を使用して、失敗したインポートの直後にこの問題を確認しました。

+0

申し訳ありません。私は他の人々のためのコード全体を投稿して、それが有用であると感じるかもしれません。 – zzj

1

REF:https://docs.python.org/3.5/extending/embedding.html

main.cppに

#include <Python.h> 
#include <iostream> 
#include <QString> 
#include <QDir> 
#include <cstring> 

using namespace std; 

int main(int argc, char *argv[]) 
{ 
    PyObject *pName, *pModule, *pDict, *pFunc; 
    PyObject *pArgs, *pValue; 
    int i; 

    if (argc < 3) { 
     fprintf(stderr,"Usage: call pythonfile funcname [args]\n"); 
     return 1; 
    } 

    Py_SetProgramName((wchar_t*)L"test"); 

    Py_Initialize(); 

    PySys_SetArgv(argc, (wchar_t**)argv); 
    PyRun_SimpleString("import tensorflow as tf\n" 
         "print(tf.__version__)\n"); 

    PyRun_SimpleString("import cv2\n" 
         "print(cv2.__version__)\n"); 

    QString qs = QDir::currentPath(); 
    std::wstring ws = qs.toStdWString(); 
    PySys_SetPath(ws.data()); 
    pName = PyUnicode_DecodeFSDefault(argv[1]); 
    /* Error checking of pName left out */ 

    pModule = PyImport_Import(pName); 
    Py_DECREF(pName); 

    if (pModule != NULL) { 
     pFunc = PyObject_GetAttrString(pModule, argv[2]); 
     /* pFunc is a new reference */ 

     if (pFunc && PyCallable_Check(pFunc)) { 
      pArgs = PyTuple_New(argc - 3); 
      for (i = 0; i < argc - 3; ++i) { 
       pValue = PyLong_FromLong(atoi(argv[i + 3])); 
       if (!pValue) { 
        Py_DECREF(pArgs); 
        Py_DECREF(pModule); 
        fprintf(stderr, "Cannot convert argument\n"); 
        return 1; 
       } 
       /* pValue reference stolen here: */ 
       PyTuple_SetItem(pArgs, i, pValue); 
      } 
      pValue = PyObject_CallObject(pFunc, pArgs); 
      Py_DECREF(pArgs); 
      if (pValue != NULL) { 
       printf("Result of call: %ld\n", PyLong_AsLong(pValue)); 
       Py_DECREF(pValue); 
      } 
      else { 
       Py_DECREF(pFunc); 
       Py_DECREF(pModule); 
       PyErr_Print(); 
       fprintf(stderr,"Call failed\n"); 
       return 1; 
      } 
     } 
     else { 
      if (PyErr_Occurred()) 
       PyErr_Print(); 
      fprintf(stderr, "Cannot find function \"%s\"\n", argv[2]); 
     } 
     Py_XDECREF(pFunc); 
     Py_DECREF(pModule); 
    } 
    else { 
     PyErr_Print(); 
     fprintf(stderr, "Failed to load \"%s\"\n", argv[1]); 
     return 1; 
    } 
    Py_Finalize(); 
    return 0; 
} 

multiply.py私はそれを解決することを

import tensorflow as tf 
import cv2 

def multiply(a,b): 
    print(tf.__version__) 
    print(cv2.__version__) 
    print("Will compute", a, "times", b) 
    c = 0 
    for i in range(0, a): 
     c = c + b 
    return c 
関連する問題