TensorFlow 2.0 C++ - 加载预训练的模型。

Question

谁能给我一个提示，我如何能用tensorflow 2.0的C++ API加载一个模型，用keras在python中训练并导出？

我找不到这方面的信息，只能用tensorflow的< 2版本。

问候

Answer 1

好吧，我找到了一个解决方案螺母与其他问题。

在Python中，你必须用以下方法导出它

tf.keras.models.save_model(model, 'model')

在C++中，你必须用..:

tensorflow::SavedModelBundle model;
tensorflow::Status status = tensorflow::LoadSavedModel(
  tensorflow::SessionOptions(), 
  tensorflow::RunOptions(), 
  "path/to/model/folder", 
  {tensorflow::kSavedModelTagServe}, 
  &model);

根据这个帖子，在C++中使用Tensorflow检查点来恢复模型 在C++中使用Tensorflow检查点来恢复模型。

如果我现在尝试设置输入和输出，它会抛出一个错误。"找不到名字为'outputlayer'的节点 "和 "无效参数，在feed_devices或fetch_devices中指定的Tensor input:0不在Graph中"。Tensor input:0, specified in either feed_devices or fetch_devices was not in the Graph".

有谁知道这里有什么问题吗？