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

北战南征 提交于 2020-06-09 17:02:25

问题


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)


回答1:


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.



来源:https://stackoverflow.com/questions/48209965/how-to-find-the-input-and-output-nodes-of-a-frozen-model

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!