Tensorflow模型移植Arm之一：C与Python互相调用

一、C调用Python

1.新建一个Python文件，名称为py_multipy.py：

#import numpy as np
def multiply(a=1,b=2):
    print('Function of python called!')
    print('a:',a)
    print('b:',b)
    print('a*b:',a*b)
    #print('numpy a*b:',np.multiply(a,b))
    

2.新建一个C调用文件，名称为call_python.c

#include <stdio.h>
#include <stdlib.h>
#include <Python.h>

int main()
{
    Py_Initialize();
    
    if(!Py_IsInitialized())
    {
        printf("Python init failed!
");
        return -1;
    }
    

    PyRun_SimpleString("import sys");
    PyRun_SimpleString("sys.path.append('./')");
       
    PyObject *pDict = NULL;
    PyObject *pModule = NULL;
    PyObject *pName = NULL;
    PyObject *pFunc = NULL;
    PyObject *pArgs = NULL;
    
    pName = PyString_FromString("py_add");
    pModule = PyImport_Import(pName);
    if (!pModule)
    {
        printf("Load py_add.py failed!
");
        getchar();
        return -1;
    }
    
    pDict = PyModule_GetDict(pModule);
    if(!pDict)
    {
        printf("Can't find dict in py_add!
");
        return -1;
    }
    
    pFunc = PyDict_GetItemString(pDict,"add");
    if(!pFunc || !PyCallable_Check(pFunc))
    {
        printf("Can't find function!
");
        getchar();
        return -1;
    }
    
    pArgs = PyTuple_New(2);
    
    PyTuple_SetItem(pArgs,0,Py_BuildValue("i",111));
    PyTuple_SetItem(pArgs,1,Py_BuildValue("i",222));
    
    PyObject_CallObject(pFunc,pArgs);
    
    if(pName)
    {
        Py_DECREF(pName);
    }
    
    if(pArgs)
    {
        Py_DECREF(pArgs);
    }
    
    if(pModule)
    {
        Py_DECREF(pModule);
    }
    
    Py_Finalize();
    return 0;

    
}

3.编译C文件

gcc -I/usr/include/python2.7/ call_python.c -o call_python -L/usr/lib/ -lpython2.7

在当前目录下生成可执行文件call_python

4.执行新生成的文件：./call_python

显示结果如下：

        Function of python called!

        ('a:',111)

        ('b:',222)

        ('a*b:',333)

二、Python调用C

1.新建一个pcallc.c

#include <stdio.h>
#include <pcallc.h>

int add(int a,int b)
{
    int c= 0;
    c = a+b;
    printf("add c:",c);
    return c;
}

2.新建一个pcallc.h

#ifndef PCALLC_H
#define PCALLC_H

int add();

#endif

3.生成动态库：pcallc.so

    gcc -o pcallc.so -shared -fPIC pcallc.c

4.新建pcallc.py

#-*- coding:utf-8 -*-

import ctypes

loadso = ctypes.cdll.LoadLibrary
lib= loadso("./pcallc.so")
lib.add(1,2)

print("***finish***")

5.运行pcallc.py

   python pcallc.py

   add c:3

   finish!