带有苗条库的Inception v3中的输入和输出节点名称是什么? [英] What are input and output node names in inception v3 with slim library?

问题描述

我使用Tensorflow slim在我自己的数据上训练了InceptionV3模型.训练后生成以下文件:-

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，检查点， events.out.tfevents

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

我想冻结图形文件并创建一个.pb文件.我不知道什么是Inception v3中的输入节点和输出节点.而且对我而言，使用Tensorboard很复杂.

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.

inceptionV3中的输入/输出节点是什么?(在slim/nets中)，或者如何找到输入/输出节点?

OS:窗口7

推荐答案

(A).如果您将其置于此链接的底部.您会在某处找到它(特定于InceptionV3):

(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).另一种方法是打印模型的所有张量并获取输入/输出张量

(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).如果需要打印.pb文件的张量名称.您可以使用此简单代码.

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

检查哪种方法对您有用.

Check what would work for you.

这篇关于带有苗条库的Inception v3中的输入和输出节点名称是什么?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！