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

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 （在笔记本的 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