使用Tensorflow图转换工具

3

当我尝试通过命令行在Windows 10中使用Graph Transform工具时,使用FrozenGraph总是会出现相同的错误。

代码:

C://tensorflow/bazel-bin/tensorflow/tools/graph_transforms/transform_graph ^
--in_graph=C://tensorflow-for-poets-2/tf_files/retrained_graph.pb ^
--out_graph=C://tensorflow-for-poets-2/new_optimized.pb ^
--inputs=Mul ^
--outputs=final_result ^
--transforms=' ^
   strip_unused_nodes ^
   fold_constants(ignore_errors=true) ^
   fold_batch_norms ^
   fold_old_batch_norms'

错误:

2018-07-27 14:59:09.303184: E tensorflow/tools/graph_transforms/transform_graph.cc:217] Unknown argument strip_unused_nodes.
usage: C://tensorflow/bazel- bin/tensorflow/tools/graph_transforms/transform_graph

我该如何操作?我需要将这个图形优化以在tensorflowjs上运行。它是InceptionV3,所以对于移动设备来说相当大。

1个回答

1

未知参数strip_unused_nodes

该错误表明缺少传递给strip_unused_nodes的参数。您需要指定输入的类型和形状。您可以查看文档


1
谢谢!事实证明,问题出在解析这个脚本上。当我使用bash来执行它们时,它能够完美地工作! - Eduardo Trunci

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接