使用 Tensorflow 图形转换工具

Question

当我尝试通过命令行将图形转换工具与 Windows 10 中的 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，所以它对于移动设备来说是相当大的。

Answer 1

Unknown argument strip_unused_nodes

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

使用 Tensorflow 图形转换工具

Using Tensorflow Graph Transformation Tools

tensorflow

tensorflow.js

tensorflowjs-converter