0

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

graph.pbtxt、model.ckpt、model.meta、model.index、检查点、events.out.tfevents

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

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

操作系统:窗口 7

4

1 回答 1

1

(一个)。如果你能做到这个链接的底部。你会在某个地方找到这个(特定于 inceptionV3):

input_layer=input
output_layer=InceptionV3/Predictions/Reshape_1

(乙)。另一种方法是打印模型的所有张量并获取输入/输出张量

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

检查什么对你有用。

于 2018-05-30T11:12:41.767 回答