如何在不使用Bazel的情况下冻结Tensorflow 1.5中的图形？

Question

在tensorflow的tools目录中，我们有冻结图python文件，我们可以直接使用它，为什么我们需要bazel，我也尝试使用该脚本，并收到以下错误：

    226   elif input_graph:
--> 227     input_graph_def = _parse_input_graph_proto(input_graph, input_binary)
    228   input_meta_graph_def = None
    229   if input_meta_graph:

/usr/local/lib/python3.5/dist-packages/tensorflow/python/tools/freeze_graph.py in _parse_input_graph_proto(input_graph, input_binary)
    167   with gfile.FastGFile(input_graph, mode) as f:
    168     if input_binary:
--> 169       input_graph_def.ParseFromString(f.read())
    170     else:
    171       text_format.Merge(f.read(), input_graph_def)

DecodeError: Error parsing message

Answer 1

您不需要bazel，只需使用适当的参数即可执行python freeze_graph.py 。

关于您的错误，看来您是在传递.pbtxt文件而不设置--input_binary=False 。 由于默认设置为True ，因此脚本希望您传递.pb文件。 当它尝试将.pbtxt解码为二进制文件时，会引发错误。