TensorFlow2.8.0代码分析之例子Lable_Image_main.cc_函数LoadGraph

函数：Status LoadGraph(const string& graph_file_name,
                 std::unique_ptr<tensorflow::Session>* session)

函数作用：从硬盘读取模型图定义，并创建可用于运行它的会话对象。。

函数声明情况如下：

 

函数声明文件如下：

 

函数流程图如下：

 

函数逻辑顺序图如下：

 

函数原始代码如下：

// Reads a model graph definition from disk, and creates a session object you
// can use to run it.
Status LoadGraph(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_name, &graph_def);
  if (!load_graph_status.ok()) {
    return tensorflow::errors::NotFound("Failed to load compute graph at '",
                                        graph_file_name, "'");
  }
  session->reset(tensorflow::NewSession(tensorflow::SessionOptions()));
  Status session_create_status = (*session)->Create(graph_def);
  if (!session_create_status.ok()) {
    return session_create_status;
  }
  return Status::OK();
}