C++调用pb文件实现模型预测

C++ 调用pb文件实现模型预测

头文件引用

 声明所需要的头文件,opencv和tensorflow头文件声明：

#include <stdio.h>
#include <vector>

#include "tensorflow/core/public/session.h"
#include "tensorflow/core/platform/env.h"

#include "opencv2/opencv.hpp"
#include "opencv2/highgui/highgui.hpp"

加载pb模型

 加载pb模型，实质上是将模型加载tensorflow图上，并到绑定定义执行环境等相关因素的session上。

• 创建图
• 模型加载
• 配置session
• session与模型图关联

// 创建图
tensorflow::GraphDef graph;
// 模型加载
tensorflow::Status ret = 
tensorflow::ReadBinaryProto(tensorflow::Env::Default(), model_path, &graph);
if (!ret.ok()) {
    std::cout << "模型加载失败" << std::endl;
   return NULL;
}
//session配置
tensorflow::Session *p_session;
NewSession(tensorflow::SessionOptions(), &p_session);
// 模型图与session相关联
ret = p_