|
由于前面的图像处理代码全是C++写的,所以我用C++调用yolov2_demo下的rknn_picture_320x320.py进行模型测试。部分代码如下:
Py_Initialize();
if (!Py_IsInitialized())
printf("Init failed.");
PyRun_SimpleString("import sys");
PyRun_SimpleString("sys.path.append('./')");
import_array();
PyObject *pModule, *pDict, *pFunction;
pModule = PyImport_ImportModule("rknn_picture_320x320");
if (pModule == NULL)
{
cout << "Python get module failed." << endl;
}
pDict = PyModule_GetDict(pModule);
一直输出Python get module failed.
我另外写了一个简单的读图像然后显示的test.py代码进行测试,C++能成功调用。但是在这个test.py中加入from rknn.api import RKNN后,又变为Python get module failed.
也测试过直接在python环境下运行rknn_picture_320x320.py,能完成测试,说明python环境中的rknn-toolkit是安装完成的。
请问在C++调用,是需要额外的什么依赖吗?
|
|