HEVC环内滤波卷积模型嵌入（2）--Python API接口

提示：C++调用Python常用接口

---

参考文章：

• C++调用Python的API总结
• 如何在C++中使用一个Python类

---

1、初始化以及结束Python接口：

Py_Initialize(); // 初始化
Py_Finalize(); // 结束关闭接口

int main(int argc, char* argv[])
{
  Py_Initialize();
  // 所有Python代码都在这个中间过程写
  Py_Finalize();
  return 0;
}

所有Python代码写在这两个接口之间

---

2、Python模型、.py文件路径

工程的当前目录指代当前工程的根目录

PyRun_SimpleString("import sys");
// 导入系统文件， sys.path 会返回系统包含路径的列表
PyRun_SimpleString("sys.path.append('./')");  
// 当前目录。指定py文件在当前路径下

“sys.path.append(‘./’)”： 把py相关文件放于工程的同一个根目录下就行
例如：

---

3、导入py文件

PyImport_ImportModule("py文件名，不需要后缀");

PyObject* pModule = PyImport_ImportModule("callModelProcess");

我的py文件如下：

这边的callModelProcess相当于模型的test.py，里面写的是模型推理测试的代码。

4、从py文件里获取所需的类函数，并执行

如：我导入的callModelProcess.py文件里有model_proc这个函数

两种加载方法

• 1、从py文件加载函数

PyObject* pFunc = NULL;
PyObject *args, *res;
pFunc= PyObject_GetAttrString(pModule, "model_proc");
// 从pModule里加载名字为model_proc的函数
args = Py_BuildValue("OO",img,37); 
//设置调用的pFunc函数的输入变量，O是变量类型格式符号，表示args是传入2个变量,一个是帧数据，一个是QP值
res = PyObject_CallObject(pFunc, args); 
//执行pFunc(args)，即model_proc(img,37),并将结果返回给res

其中C++向Python传递值函数
Py_BuildValue("(i)",256);是C类型的数据结构转换成Python对象，函数原型PyObject *Py_BuildValue(char *format, ...)
所传变量的类型格式符号对照：

变量格式符
例如

input:                                         return:
Py_BuildValue("s", "hello")                            'hello'
Py_BuildValue("ss", "hello", "world")                  'hello', 'world'
Py_BuildValue("s#", "hello", 4)                        'hell'
Py_BuildValue("i", 123)                                123
Py_BuildValue("iii", 123, 456, 789)                    123, 456, 789
Py_BuildValue("()")                                    ()
Py_BuildValue("")                                      None
Py_BuildValue("(i)", 123)                              (123,)
Py_BuildValue("(ii)", 123, 456)                        (123, 456)
Py_BuildValue("(i,i)", 123, 456)                       (123, 456)
Py_BuildValue("[i,i]", 123, 456)                       [123, 456]
Py_BuildValue("{s:i,s:i}", "abc", 123, "def", 456)     {'abc': 123, 'def': 456}
Py_BuildValue("((ii)(ii)) (ii)", 1, 2, 3, 4, 5, 6)     (((1, 2), (3, 4)), (5, 6))

创建一个元组，可以同时传递多个变量
PyObject *args = PyTuple_New(3); // 创建有3个元素的元组
PyObject* arg1 = Py_BuildValue("i", 100); // 整数参数
PyObject* arg2 = Py_BuildValue("f", 3.14); // 浮点数参数

PyTuple_SetItem(args, 0, arg1);  // 元组第一个添加为arg1
PyTuple_SetItem(args, 1, arg2);
PyTuple_SetItem(args, 2, Py_BuildValue("s", "hello")); // 元组第3个元素为"hello"

多个变量传递，可以简化为如下形式
PyObject* args = Py_BuildValue("(ifs)", 100, 3.14, "hello");

• py文件加载类，从类中加载函数
  假设"model_proc"是一个我们自己定义的python类

PyObject* pDict = NULL;
PyObject* pClass = NULL;
PyObject *args, *res, *pInstance;
PyObject* pModule = PyImport_ImportModule("callModelProcess");
pDict = PyModule_GetDict(pModule); 
// 获取pModule中的所有类和定义的函数
pClass = PyDict_GetItemString(pDict, "model_proc");
// 获取名为model_proc的类或者函数
args = Py_BuildValue("(s)","./config.txt"); 
//设置类初始化需要的参数
pInstance=PyInstance_New(pClass, class_args, NULL ); 
//初始化"model_proc"，建立实例pInstance
res=PyObject_CallMethod(pInstance,"func","(i)",12345); 
// 执行pInstance.func(12345)，即执行Python类里面的函数

---

class_args = Py_BuildValue(“(s)”,“./config.txt”);
传参方式还可以直接整个txt文件

---

5、Python返回值传递给C++

PyArg_ParseTuple() 返回多个值
PyArg_Parse() 返回单个值

PyArg_ParseTuple()函数的定义:

int PyArg_ParseTuple(PyObject *res, char *format, &c++地址);

int i_c;
char *s_c
PyArg_ParseTuple(res, "is", &i_c, &s_c);
// 将Python返回的res中的两个值，解析后，传给C++中整形i_c，和字符串s_c的地址

---

• CSDN 技术博客 2 篇
  2023.03.30