转换pb_使用tf的c++API加载pb做前向推理

背离赤道逆光而行

于 2020-12-29 07:22:48 发布

阅读量117

点赞数

文章标签：转换pb

本文链接：https://blog.csdn.net/weixin_35257663/article/details/112742892

版权

本文介绍如何使用TensorFlow的C++ API加载.pb模型文件进行前向推理，包括加载模型、指定起止节点进行前向运算，并提供了一段封装后的伪代码示例。

摘要由CSDN通过智能技术生成

前面我们介绍了将tensorflow的ckpt模型文件转换为pb文件，接下来，我们使用tensorflow的C++接口调用pb文件做前向推理。

这里假设已经编译好了tensorflow的动态库

加载模型

加载模型的过程和python接口时一致，先创建一个session会话，定义一个graph，读取pb文件后使用session为定义的graph分配资源。

using tensorflow::Flag;
using tensorflow::Tensor;
using tensorflow::Status;
using tensorflow::string;
using tensorflow::int32;
using tensorflow::uint8;
using namespace tensorflow;
using namespace std;
using namespace cv;

Status load_graph(const string& graph_file_name, std::unique_ptr<tensorflow::Session>* session) {
    
    tensorflow::GraphDef graph_def;
    Status load_graph_status = ReadBinaryProto(tensorflow::Env::Default(), graph_file_nam