手机版
你好,游客 登录 注册
背景:
阅读新闻

C++调用Python中的Tensorflow模型

[日期:2019-03-02] 来源:Linux社区  作者:TangCode [字体: ]

利用c++调用Python2.7的程序,加载tensorflow模型(为什么不使用Python3,坑太多了,一直解决不好)。整个环境在Ubuntu16.04下完成,利用了kDevelop4 IDE编写C++程序,以及cmake文件。

保存tensorflow模型

首先利用Python写一段tensorflow保存模型的代码:

import tensorflow as tf
import os

def save_model_ckpt(ckpt_file_path):
    x = tf.placeholder(tf.int32,name='x')
    y = tf.placeholder(tf.int32,name='y')
    b = tf.Variable(1,name='b')
    xy = tf.multiply(x,y)
    op = tf.add(xy,b,name='op_to_store')
    sess = tf.Session()
    sess.run(tf.global_variables_initializer())

    path = os.path.dirname(os.path.abspath(ckpt_file_path))
    if os.path.isdir(path) is False:
        os.makedirs(path)

    tf.train.Saver().save(sess,ckpt_file_path)

    feed_dict = {x:4,y:3}
    print(sess.run(op,feed_dict))

save_model_ckpt('./model/model.ckpt')

这会在model目录下回保存四个文件

模型加载代码

#classify.py
import tensorflow as tf

def evaluate(pic): 
    sess = tf.Session()
    saver = tf.train.import_meta_graph('/home/tyl/Code/Kprojects/cpython/Test/model/model.ckpt.meta')
    saver.restore(sess, tf.train.latest_checkpoint('../model'))
    print(type(sess.run('b:0')))
    input_x = sess.graph.get_tensor_by_name('x:0')
    input_y = sess.graph.get_tensor_by_name('y:0')
    op = sess.graph.get_tensor_by_name('op_to_store:0')
    add_on_op = tf.multiply(op,2)
    ret = sess.run(add_on_op,{input_x:5,input_y:5})
    print ret
    sess.close()
    return pic

这里要注意的是模型加载的路径一定要正确。。。。
C++程序调用Python程序
这里,利用C++程序调用模型加载的Python程序

//readTF.cpp
#include <Python.h>
#include <pythonrun.h>
#include <iostream>
#include <string.h>

int main()
{
  const int flag= 1;
  Py_Initialize();
  if (!Py_IsInitialized())
  {
    return -1;
  }

  PyRun_SimpleString("import sys");
  //路径一定要对
  PyRun_SimpleString("sys.path.append('/home/tyl/Code/Kprojects/cpython/Test')");
 
  PyObject* pMod = NULL;
  PyObject* pFunc = NULL;
  PyObject* pParm = NULL;
  PyObject* pRetVal = NULL;
  int iRetVal=999;
  PyObject* pName = PyString_FromString("classify");
  pMod = PyImport_Import(pName);//获取模块
  if (!pMod)
  {
 std::cout << pMod <<std::endl;
    return -1;
  }
  const char* funcName = "evaluate";
  pFunc = PyObject_GetAttrString(pMod,funcName);//获取函数
  if (!pFunc)
  {
    std::cout << "pFunc error" <<std::endl;
    return -1;
  }
 
  pParm = PyTuple_New(1);//新建元组
  PyTuple_SetItem(pParm, 0, Py_BuildValue("i",flag));//向Python模块传参
  pRetVal = PyObject_CallObject(pFunc,pParm);//获得返回结果

  PyArg_Parse(pRetVal,"i",&iRetVal);//解析成C++需要的形式
  std::cout<< iRetVal <<std::endl;
  return 0;
}

CMakeLists文件书写
cmake_minimum_required(VERSION 2.6)
project(test)
set (CMAKE_BUILD_TYPE Debug)
set (CMAKE_CXX_FLAGS "-std=c++11")

include_directories( /usr/include/python2.7)     
add_executable(readTF readTF.cpp)
target_link_libraries(readTF -lpython2.7)

结果
在KDevelop4上运行的结果

C++调用Python中的Tensorflow模型

Linux公社的RSS地址https://www.linuxidc.com/rssFeed.aspx

本文永久更新链接地址https://www.linuxidc.com/Linux/2019-03/157201.htm

linux
相关资讯       C++调用Python  Tensorflow模型 
本文评论   查看全部评论 (0)
表情: 表情 姓名: 字数

       

评论声明
  • 尊重网上道德,遵守中华人民共和国的各项有关法律法规
  • 承担一切因您的行为而直接或间接导致的民事或刑事法律责任
  • 本站管理人员有权保留或删除其管辖留言中的任意内容
  • 本站有权在网站内转载或引用您的评论
  • 参与本评论即表明您已经阅读并接受上述条款