TensorFlow saved_model对于不同的输入返回相同的结果

时间:2019-04-17 20:22:02

标签: tensorflow tensorflow-serving

我正在尝试使用以下代码导出保存的模型。它导出了saved_model.pb,但是对于我提供的任何输入,我都得到相同的结果。我认为我的输出方式有问题。我是TensorFlow的新手,如果这是一个简单的错误,我深表歉意。

with tf.gfile.FastGFile("retrained_graph.pb", 'rb') as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())
    _ = tf.import_graph_def(graph_def, name='')

def export_model(saved_model_dir, final_tensor_name):
  with tf.Session() as sess:
    init = tf.global_variables_initializer()
    sess.run(init)
    with sess.graph.as_default() as graph:
      tf.saved_model.simple_save(
        sess,
        saved_model_dir,
        inputs={'image': tf.placeholder(tf.string)},
        outputs={'prediction': graph.get_tensor_by_name(final_tensor_name + ":0")}
   )

我还使用了以下教程中的retrain.py:https://github.com/BartyzalRadek/Multi-label-Inception-net,因此可能来自那里。

我如何构造代码,是因为我将其包含在单独的文件中,因此无需每次尝试保存时都进行重新训练。

0 个答案:

没有答案