C++에서 파이썬을 임베드 할 때 tensorflow를 가져 오면 null이 반환됩니다.

Question

파이썬을 C++ 응용 프로그램에 임베드하는 것과 관련된 질문이 있습니다. 설정은 다음과 같습니다 : 나는 어떤 데이터 (실시간으로 이미지를 렌더링)를 생성하고 그들을 표시하는 대형 C++ 응용 프로그램이 있습니다. 나는 또한 이러한 이미지를 받아들이는 tensorflow를 사용하여 파이썬에서 신경망을 훈련시켰다.

제 아이디어는 파이썬을 임베드하고 데이터를 숫자가 적은 배열로 보내고, 신경망을 사용하여 예측하고 표시 할 다른 처리 된 배열을 다시 가져 오는 것입니다 (C++에서). 파이썬을 C 언어로 임베딩 한 느낌을 얻기 위해 파이썬 측에 tensorflow가없는 몇 가지 기본 테스트를 만들었습니다.

그러나 "import tensorflow"를 가져 오려는 파이썬 스크립트에 넣으면 C++ 부분의 PyImport_ImportModule에서 NULL을 가져옵니다.

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

잘 작동합니다. 그러나 다음은하지 않습니다

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에 의지 할 것입니다.

Answer 1

문제가 해결되었습니다. argS와 argv를 PySys_SetArgv로 설정해야했습니다. 문제점을보기 위해 실패한 가져 오기 직후에 PyErr_Occurred() 및 PyErr_Print()를 사용하여 이것을 밝혀 냈습니다.

Answer 2

심판 : 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