使用Python导出Tensorflow网络并在C+中导入+;没有巴泽尔 使用TensorFlow,我试图在C++文件中加载一个网络,我用Python进行了训练。在Python上,我正在使用输入张量x和输出张量y保存一个网络: with tf.Session(graph=tf.Graph()) as sess: tf.saved_model.simple_save(sess, export_dir, inputs={"ob": x}, outputs={"out": y})
接下来,我将在我的C文件中执行以下操作:使用Python导出Tensorflow网络并在C+中导入+;没有巴泽尔 使用TensorFlow,我试图在C++文件中加载一个网络,我用Python进行了训练。在Python上,我正在使用输入张量x和输出张量y保存一个网络: with tf.Session(graph=tf.Graph()) as sess: tf.saved_model.simple_save(sess, export_dir, inputs={"ob": x}, outputs={"out": y}),python,c++,tensorflow,Python,C++,Tensorflow,接下来,我将在我的C文件中执行以下操作: TF_Buffer* buffer = ReadBufferFromFile("graph.pb"); TF_Graph* graph = TF_NewGraph(); TF_Status* status = TF_NewStatus(); TF_ImportGraphDefOptions* opts = TF_NewImportGraphDefOptions(); TF_GraphImportGraphDef(graph, buffer, opts,
TF_Buffer* buffer = ReadBufferFromFile("graph.pb");
TF_Graph* graph = TF_NewGraph();
TF_Status* status = TF_NewStatus();
TF_ImportGraphDefOptions* opts = TF_NewImportGraphDefOptions();
TF_GraphImportGraphDef(graph, buffer, opts, status);
但是在执行TF\u GraphImportGraphDef
调用时,我得到了一个TF\u Status
error codeTF\u INVALID\u参数。正在加载缓冲区,我可以打印它的长度并获得一个值(本例中为192)
我应该怎么做,或者怎么做?使用BAZEL和C++ API不是一个选项。是否在TysFROUM 2中尝试?@ JRealAlad NoPE,它没有使它与TfLoxSeStReSaveDead模型一起工作,并且可以将模型训练成将权值保存到检查点文件。但是我可以;我无法再次将上述检查点重新加载到python中。它快把我逼疯了