Question

我是Tensorflow的新手，并尝试运行位于https://github.com/Hvass-Labs/TensorFlow-Tutorials/blob/master/02_Convolutional_Neural_Network.ipynb的一个教程代码

基于此代码，我想尝试使用simple_save以.pb格式保存模型并将其还原以进行测试，但是我不知道如何修改这段代码。我浏览了一些网页，但还是不明白。谁能帮助我更改这段代码，以便我可以保存经过训练的模型，然后加载以进行推理？谢谢！

Answer 1

要保存模型，您需要做两件事-输入和输出张量名称。在您的情况下，输入张量称为x，输出张量称为y_pred和y_pred_cls（在notebook的In [29]中提到）。这是保存模型的简单示例：

simple_save(session,
            export_dir,
            inputs={"x": x,},
            outputs={"y_pred": y_pred,
                     "y_pred_cls": y_pred_class})

编辑：恢复-

restoring_graph = tf.Graph()
with restoring_graph.as_default():
    with tf.Session(graph=restoring_graph) as sess:
       # Restore saved values
       tf.saved_model.loader.load(
          sess,
          [tag_constants.TRAINING],
          export_dir  # Path to SavedModel
       )
      # Pass inputs to model and do predictions below

如何以.pb格式保存模型，然后在Tensorflow中加载模型以进行推理？

1 个答案: