使用Tensorflow图转换工具

时间:2018-07-27 18:02:04

标签: tensorflow tensorflow.js tensorflowjs-converter

当我尝试通过命令行将图形转换工具与Windows 10中的FrozenGraphic一起使用时,总是出现相同的错误。 代码:

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 个答案:

答案 0 :(得分:1)

  

未知参数strip_unused_nodes

该错误表明strip_unused_nodes中缺少参数。您需要指定输入的类型和形状。您可以看看docs