TensorFlow freeze_graph：需要位置参数‘unused_args’

Question

我在带有 High Sierra 的 Mac 上使用带有 Python 3.6.5 的 TensorFlow 1.7。

我训练了我的第一个 MNIST 模型，所以我基本上

一个 graph.pbtxt 具有 CNN 图结构的文件
一些 model.ckpt-21000 个文件（.meta、.index、.data）

我试图在 bash:

上使用命令行 freeze_graph 命令冻结图表

freeze_graph
--input_graph=/…/graph.pbtxt
--input_checkpoint=/…/model.ckpt-21000
--input_binary=false
--output_graph=/…/frozen_mnist.pb
--output_node_names=softmax_tensor

但是我得到了这个错误：

Traceback (most recent call last):  
File “/usr/local/bin/freeze_graph”, line 11, in <module>  
sys.exit(main())  
TypeError: main() missing 1 required positional argument: ‘unused_args’

我不太确定我在那里遗漏了什么。我很确定我使用的是正确的语法。

Answer 1

我找到了冻结图表的解决方法。我把它贴在这里，所以如果有人遇到同样的问题，他们可以使用它。

而不是

freeze_graph
--input_graph=/…/graph.pbtxt
--input_checkpoint=/…/model.ckpt-21000
--input_binary=false
--output_graph=/…/frozen_mnist.pb
--output_node_names=softmax_tensor

使用

python3 -m tensorflow.python.tools.freeze_graph
--input_graph=/…/graph.pbtxt
--input_checkpoint=/…/model.ckpt-21000
--input_binary=false
--output_graph=/…/frozen_mnist.pb
--output_node_names=softmax_tensor

所以基本上我只是使用 python3 -m tensorflow.python.tools.freeze_graph[=24= 而不是命令 freeze_graph ].

我仍然很想了解为什么命令行对我不起作用:(

TensorFlow freeze_graph：需要位置参数‘unused_args’

TensorFlow freeze_graph: required positional argument ‘unused_args’

python

mnist

tensorflow