python freeze_graph.py --input_graph=model.pb --input_checkpoint=model.ckpt --output_graph=asrModel.pb --output_node_names=dense_2/Softmax --input_binary=True
得到{{output_node_names}},我们可以在model得到output_node_names,我们可以在model.py中添加一句代码,然后找到相应输出即可。
...
#x = Dropout(DROP_RATE)(x) x = Dense(24, activation='softmax')(x) print(x.name) #add this line model = Model(input=In, output=x)
...