What are input and output node names in inception v3 with slim library?

╄→гoц情女王★ 提交于 2020-01-25 06:41:08

问题


I retrained inceptionV3 model on my own data using Tensorflow slim. Below files are generated after training :-

graph.pbtxt, model.ckpt, model.meta, model.index, checkpoint, events.out.tfevents

I want to freeze the graph files and create a .pb file. I don't know what is input node and output node in inception v3. And using Tensorboard is complex for me.

What are the input/output nodes in inceptionV3?(in slim/nets) OR how can I find the input/output nodes ?

OS : window 7


回答1:


(A). If you will make it to bottom of this link. You would find this somewhere(specific to inceptionV3) :

input_layer=input
output_layer=InceptionV3/Predictions/Reshape_1

(B). Another way is to print all tensors of the model and get input/output tensor

from tensorflow.python.tools.inspect_checkpoint import print_tensors_in_checkpoint_file
ckpt_path="model.ckpt"
print_tensors_in_checkpoint_file(file_name=ckpt_path, tensor_name='', all_tensors=True, all_tensor_names=True)

(C). If you need to print tensor names of .pb file. You can use this simple code.

Check what would work for you.



来源:https://stackoverflow.com/questions/50595658/what-are-input-and-output-node-names-in-inception-v3-with-slim-library

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