有没有办法在C ++中保存冻结的tensorflow图？

Question

我知道您可以使用Python将模型保存到检查点或SavedModel中。我想知道是否有一种方法可以使用C ++保存从FreezeSavedModel C ++函数返回的GraphDef。谢谢！

Answer 1

函数FreezeSavedModel返回 GraphDef -Object。您可以使用 WriteBinaryProto 或 WriteTextProto

保存该对象

//BinaryProto
const string binary_file = io::JoinPath(testing::TmpDir(), "binary_graph.pb");
TF_ASSERT_OK(WriteBinaryProto(Env::Default(), binary_file, graph_def));

//Textproto
const string text_file = io::JoinPath(testing::TmpDir(), "text_graph.pbtxt");
TF_ASSERT_OK(WriteTextProto(Env::Default(), text_file, graph_def));

您可以在file_utils_test.cc

中找到示例