带有 slim 库的 inception v3 中的输入和输出节点名称是什么？

Question

我使用 Tensorflow slim 在我自己的数据上重新训练了 inceptionV3 模型。以下文件是训练后生成的：-

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

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

inceptionV3 中的 input/output 节点是什么？（在 slim/nets 中）或者如何找到 input/output 节点？

OS : window 7

Answer 1

(A)。如果你能到达 this link 的底部。你会在某处找到这个（特定于 inceptionV3）：

input_layer=input
output_layer=InceptionV3/Predictions/Reshape_1

(乙)。另一种方法是打印模型的所有张量并得到input/output张量

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 文件的张量名称。您可以使用简单代码。

检查什么适合您。

带有 slim 库的 inception v3 中的输入和输出节点名称是什么？

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

tensorflow

tensorflow-slim