tensorflow - 使用 Tensorflow 图形转换工具

标签 tensorflow tensorflow.js tensorflowjs-converter

当我尝试通过命令行在 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,因此对于移动设备来说相当大。

最佳答案

Unknown argument strip_unused_nodes

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

关于tensorflow - 使用 Tensorflow 图形转换工具,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51563280/

相关文章:

javascript - 如何使用 Tensorflow.js 计算张量中每个值出现的次数?

tensorflow - 我转换后的 tensorflow 迁移学习模型总是在 Tensorflow JS 中返回相同的结果

python - 将稳定基线 tensorflow 模型转换为 tensorflow js

javascript - 需要找出需要为 Tensorflowjs 转换器传递哪些输出节点名称

python - 如何在keras中拟合模型以解决多输入问题

tensorflow - 为什么我们必须在 tensorflow 的反卷积过程中指定输出形状?

javascript - 继续获取 : "Attempted import error: ' loadFrozenModel' is not exported from '@tensorflow/tfjs-converter' "

python - 将 report_tensor_allocations_upon_oom 添加到 cifar10_estimator 示例

electron - 带TensorflowJS的 Electron -寻找napi-v4,但仅存在napi-v5

javascript - 将 png 转换为张量 tensorflow.js