当前位置: 首页 > 知识库问答 >
问题:

Tensorflow:如何从模型图中获取输入和输出节点?

柳浩大
2023-03-14

我用Tensorflow培训了一个模型(更快的rcnn\U resnet101\U coco\U 2018\U 01\U 28)。我有一个“.pb”图形。

要制作一个冻结图,我需要输入和输出节点。

如何在图表中找到它?

我这里有完整的节点列表。

没有任何节点像Softmax,占位符,因为它在其他帖子中建议。

共有1个答案

段干昊然
2023-03-14

使用C构建的汇总图。我的建议是,您可以使用Docker运行Summary graph并获取经过训练的模型的输入和输出名称。请参阅此链接

图形转换工具的GITHUB链接(汇总图形)

bazel build tensorflow/tools/graph\u transforms:summary\u graph
bazel bin/tensorflow/tools/graph\u transforms/summary\u graph--in\u graph=tensorflow\u inception\u graph。pb

 类似资料:
  • 我训练了一个模型,输入了张量:输入形状:[150]输入类型: 在输出中,该张量为:输出形状:[1 50 17]输出类型: 如何在Java中创建这些?在Android设备上使用它?谢谢

  • 当我将张量流精简模型添加到我的Android应用程序时。它建议自动生成的代码。 现在让我们假设我在python中的输入形状是一个50数字[1,2,3...]的int数组,它给出了一个浮点值的输出。 我必须以何种方式更改代码。

  • 本文向大家介绍TensorFlow查看输入节点和输出节点名称方式,包括了TensorFlow查看输入节点和输出节点名称方式的使用技巧和注意事项,需要的朋友参考一下 TensorFlow 定义输入节点名称input_name: TensorFlow查看pb数据库里面的输入节点和输出节点: 以上这篇TensorFlow查看输入节点和输出节点名称方式就是小编分享给大家的全部内容了,希望能给大家一个参考,

  • 我有一个EXE文件,< code>addOne.exe,它不断地从控制台上的用户获得一个整数输入(不是命令行参数),并将整数1输出到控制台上。示例输出如下所示: 我正在尝试编写一个java程序,它可以: 运行 EXE 使用 从 Java 程序持续获取用户输入,并将输入到 EXE 作为控制台输入 每当 EXE 将文本输出到控制台时,请从 Java 程序打印该文本 我可以使用以下方式运行EXE: 但我

  • 我使用以下代码生成量化的tflite模型 但是根据训练后量化: 生成的模型将完全量化,但为了方便起见,仍然采用浮点输入和输出。 要为Google Coral Edge TPU编译tflite模型,我还需要量化输入和输出。 在模型中,我看到第一个网络层将浮点输入转换为,最后一个网络层将转换为浮点输出。如何编辑tflite模型以除去第一个和最后一个浮动层? 我知道我可以在转换期间将输入和输出类型设置为