我正在尝试使用以下代码导出保存的模型。它导出了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,因此可能来自那里。
我如何构造代码,是因为我将其包含在单独的文件中,因此无需每次尝试保存时都进行重新训练。