How to find the Input and Output Nodes of a Frozen Model

tensorflow inference object-detection-api
gustavz picture gustavz · Jan 11, 2018 · Viewed 12.2k times · Source

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?

Here is a link to the graph generated by tensorboard

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)

Answer

Pete Warden picture Pete Warden · Feb 5, 2018

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

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

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

Related questions

TensorFlow Lite C++ API example for inference

I am trying to get a TensorFlow Lite example to run on a machine with an ARM Cortex-A72 processor. Unfortunately, I wasn't able to deploy a test model due to the lack of examples on how to use the C++ …

Read More
How to find which version of TensorFlow is installed in my system?

I need to find which version of TensorFlow I have installed. I'm using Ubuntu 16.04 Long Term Support.

Read More
TensorFlow not found using pip

I'm trying to intstall TensorFlow using pip: $ pip install tensorflow --user Collecting tensorflow Could not find a version that satisfies the requirement tensorflow (from versions: ) No matching distribution found for tensorflow What am I doing wrong? So far I've used …

Read More