我已经使用Bazel从源代码构建了TensorFlow,但是,当我尝试运行freeze_graph脚本时,我收到请求.runfiles的错误
下面是我的run命令
python /home/olu/TensorFlow_Source/tensorflow/bazel-bin/tensorflow/python/tools/freeze_graph --input_graph=graph.pbtx --input_checkpoint=model.ckpt-0.data-00000-of-00001 --input_binary=false --output_graph=../frozen_graph/frozen_graph.pb --output_node_names=ArgMax
下面是它生成的错误:
Traceback (most recent call last):
File "/home/olu/TensorFlow_Source/tensorflow/bazel-bin/tensorflow/python/tools/freeze_graph", line 172, in <module>
Main()
File "/home/olu/TensorFlow_Source/tensorflow/bazel-bin/tensorflow/python/tools/freeze_graph", line 111, in Main
module_space = FindModuleSpace()
File "/home/olu/TensorFlow_Source/tensorflow/bazel-bin/tensorflow/python/tools/freeze_graph", line 86, in FindModuleSpace
raise AssertionError('Cannot find .runfiles directory for %s' % sys.argv[0])
AssertionError: Cannot find .runfiles directory for /home/olu/TensorFlow_Source/tensorflow/bazel-bin/tensorflow/python/tools/freeze_graph
你知道我可能做错了什么吗?
发布于 2018-08-13 04:28:17
在我的例子中,我得到了相同的错误,我需要首先构建freeze_graph:
bazel build tensorflow/python/tools:freeze_graph
在此之后,可以这样运行它:
bazel-bin/tensorflow/python/tools/freeze_graph \ --input_graph=/tmp/inception_v3_inf_graph.pb \ --input_checkpoint=/tmp/checkpoints/inception_v3.ckpt \ --input_binary=true --output_graph=/tmp/frozen_inception_v3.pb \ --output_node_names=InceptionV3/Predictions/Reshape_1
大卫
https://stackoverflow.com/questions/48081738
复制相似问题