Tensorflow无法运行strip_unused.pb（没有属性'__path__'）

Question

遵循TensorFlow for Poets（Android）（https://codelabs.developers.google.com/codelabs/tensorflow-for-poets-2/#0）

的教程

• 尝试使用Inception模型而不是Mobilenet

尝试使用strip_unused.py从重新训练的模型中剥离DecodeJpeg Op，但遇到以下错误。

错误：

/home/user/tensorflow/bin/python: Error while finding spec for 'tensorflow.python.tools.strip_unused.py' (AttributeError: module 'tensorflow.python.tools.strip_unused' has no attribute '__path__')

命令行：

python -m tensorflow.python.tools.strip_unused.py --input_graph=tf_files/retrained_graph.pb --output_graph=tf_files/stripped_graph.pb  --input_node_names="Mul"  --output_node_names="final_result" --input_binary=true

机：

• Ubuntu 16.04 LTS
• Python 3.5.2
• TensorFlow 1.4.1

非常感谢任何帮助。谢谢！

Answer 1

可能由于拼写错误，因为指定了文件扩展名.py。这似乎有效：

python -m tensorflow.python.tools.strip_unused --input_graph=tf_files/retrained_graph.pb --output_graph=tf_files/stripped_graph.pb  --input_node_names="Mul"  --output_node_names="final_result" --input_binary=true

结果：最终图表中有997个操作。