我使用Tensorflow slim在我自己的数据上重新训练了inceptionV3模型。下面的文件是在培训后生成的: -
graph.pbtxt,model.ckpt,model.meta,model.index,checkpoint, events.out.tfevents
我想冻结图形文件并创建.pb
文件。我不知道在初始v3中输入节点和输出节点是什么。使用Tensorboard对我来说很复杂。
inceptionV3中的输入/输出节点是什么?(在slim / nets中)或者我如何找到输入/输出节点?
操作系统:窗口7
答案 0 :(得分:1)
(A)。如果你将它放到this link的底部。你会在某个地方找到它(特定于inceptionV3):
input_layer=input
output_layer=InceptionV3/Predictions/Reshape_1
(B)。另一种方法是打印模型的所有张量并获得输入/输出张量
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文件的张量名称。您可以使用this简单代码。
检查什么对你有用。
答案 1 :(得分:0)
使用slim.get_variables_to_restore()
。您可以找到详细信息here。