Toybrick

标题: 关于模型测试问题 [打印本页]

作者: yangyi129 时间: 2020-3-31 16:41
标题: 关于模型测试问题
由于前面的图像处理代码全是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++调用，是需要额外的什么依赖吗？

作者: jefferyzhang 时间: 2020-3-31 17:00
没这么玩过。。
没有其他依赖了，c++调用直接用rknn_api就好了，没几行代码
rknn_api里也有对应sample

欢迎光临 Toybrick (https://t.rock-chips.com/)