Tensorflow freeze_graph脚本在使用Keras定义的模型上失败

Question

我正在尝试将使用Keras构建和训练的模型导出到我可以在C++脚本中加载的protobuffer(如本例所示).我生成了一个包含模型定义的.pb文件和一个包含检查点数据的.ckpt文件.但是,当我尝试使用freeze_graph脚本将它们合并到一个文件中时,我收到错误:

ValueError: Fetch argument 'save/restore_all' of 'save/restore_all' cannot be interpreted as a Tensor. ("The name 'save/restore_all' refers to an Operation not in the graph.")

我正在保存这样的模型:

with tf.Session() as sess:
    model = nndetector.architecture.models.vgg19((3, 50, 50))
    model.load_weights('/srv/nn/weights/scratch-vgg19.h5')
    init_op = tf.initialize_all_variables()
    sess.run(init_op)
    graph_def = sess.graph.as_graph_def()
    tf.train.write_graph(graph_def=graph_def, logdir='.',   name='model.pb', as_text=False)
    saver = tf.train.Saver()
    saver.save(sess, 'model.ckpt')

nndetector.architecture.models.vgg19((3,50,50))只是一个在Keras中定义的类似vgg19的模型.

我正在调用freeze_graph脚本:

bazel-bin/tensorflow/python/tools/freeze_graph --input_graph=[path-to-model.pb] --input_checkpoint=[path-to-model.ckpt] --output_graph=[output-path] --output_node_names=sigmoid --input_binary=True

如果我运行freeze_graph_test脚本一切正常.

有谁知道我做错了什么？

谢谢.

最好的祝福

菲利普

编辑

我试过打印tf.train.Saver().as_saver_def().restore_op_name返回save/restore_all.

另外,我尝试了一个简单的纯tensorflow示例,仍然得到相同的错误:

a = tf.Variable(tf.constant(1), name='a')
b = tf.Variable(tf.constant(2), name='b')
add = tf.add(a, b, 'sum')

with tf.Session() as sess:
sess.run(tf.initialize_all_variables())
tf.train.write_graph(graph_def=sess.graph.as_graph_def(), logdir='.',     name='simple_as_binary.pb', as_text=False)
tf.train.Saver().save(sess, 'simple.ckpt')

而且我实际上也无法在python中恢复图形.ValueError: No variables to save如果我单独执行它来保存图形,则使用以下代码抛出(也就是说,如果我在同一个脚本中保存和恢复模型,一切正常).

with gfile.FastGFile('simple_as_binary.pb') as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())

with tf.Session() as sess:
    tf.import_graph_def(graph_def)
    saver = tf.train.Saver()
    saver.restore(sess, 'simple.ckpt')

我不确定这两个问题是否相关,或者我是不是在python中没有正确恢复模型.

Answer 1

问题是原始程序中这两行的顺序:

tf.train.write_graph(graph_def=sess.graph.as_graph_def(), logdir='.',     name='simple_as_binary.pb', as_text=False)
tf.train.Saver().save(sess, 'simple.ckpt')

调用tf.train.Saver() 会向图中添加一组节点,包括一个节点"save/restore_all".但是,此程序在写出图形后调用它,因此传递给的文件freeze_graph.py不包含执行重写所必需的那些节点.

反转这两行应该使脚本按预期工作:

tf.train.Saver().save(sess, 'simple.ckpt')
tf.train.write_graph(graph_def=sess.graph.as_graph_def(), logdir='.',     name='simple_as_binary.pb', as_text=False)