【问题标题】:TensorFlow freeze_graph: required positional argument ‘unused_args’TensorFlow freeze_graph:必需的位置参数“unused_args”
【发布时间】:2018-04-06 08:36:42
【问题描述】:

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

我已经训练了我的第一个 MNIST 模型,所以我基本上有

  • 具有 CNN 图结构的 graph.pbtxt 文件
  • 一些 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’

我不确定我在那里缺少什么。 我很确定我使用了正确的语法。

【问题讨论】:

    标签: python tensorflow mnist


    【解决方案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 而不是命令 freeze_graph

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

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 2013-05-17
      • 2021-04-07
      • 1970-01-01
      • 2014-10-14
      • 1970-01-01
      • 2019-12-01
      • 2018-09-09
      相关资源
      最近更新 更多