如何找到冻结模型的输入和输出节点 [英] How to find the Input and Output Nodes of a Frozen Model

问题描述

我想在来自模型动物园的冻结模型上使用 tensorflow 的 optimize_for_inference.py 脚本:ssd_mobilenet_v1_coco.

I want to use tensorflow's optimize_for_inference.py script on a frozen Model from the model zoo: the ssd_mobilenet_v1_coco.

如何查找/确定模型的输入名称和输出名称?

How do i find/determine the names of the input and output name of the model?

这里是张量板生成的图的链接

这个问题可能有帮助:给定张量流模型图，如何找到输入节点和输出节点名称(对我来说没有)

This question might help: Given a tensor flow model graph, how to find the input node and output node names (for me it did not)

推荐答案

我们现在已经添加了一些关于这个过程的文档，这里有:

We've now added some documentation on this process, covered here:

https://www.tensorflow.org/mobile/prepare_models

如果您查找 summary_graph，您将看到一个示例，说明如何找出正确的输入和输出节点.

If you look for summarize_graph, you'll see an example of how to figure out the right input and output nodes.

这篇关于如何找到冻结模型的输入和输出节点的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！