如何在不使用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解碼為二進制文件時，會引發錯誤。