如何找到output_node_name来构建冻结图?

我参考了https://github.com/dennybritz/cnn-text-classification-tf作为参考。我的目标是从模型文件中构建冻结图。我想知道签名中的输入和输出节点,以便有效地构建冻结图。我使用以下代码从图定义中打印proto文件。

saver=tf.train.import_meta_graph('some_path/model.ckpt.meta')imported_graph = tf.get_default_graph() graph_op = imported_graph.get_operations() with open('output.txt', 'w') as f:        for i in graph_op:            f.write(str(i))

我得到的输出如下:https://drive.google.com/drive/folders/1iZQqohx8jAWbSw7XV3vFJuLkaUp0Dt2s?usp=sharing

如何知道哪个是输出节点,哪个是输入节点?这个文件中有大量的输入和输出?


回答:

我建议使用Tensorboard来可视化图结构,而不是使用带有节点的文本文件。你可以在这里找到更多细节 这里

然而,图本身并没有输入或输出的概念。你可以将没有输入连接的节点视为可能是输入节点,特别是占位符节点。连接到损失函数的节点是可能是输出节点的候选者。

总结:一般来说,你需要通过分析网络架构来猜测哪些节点是输入,哪些是输出。

至于你提到的仓库,你可以阅读eval.py代码,并找到这些行:

input_x = graph.get_operation_by_name("input_x").outputs[0]# input_y = graph.get_operation_by_name("input_y").outputs[0]dropout_keep_prob = graph.get_operation_by_name("dropout_keep_prob").outputs[0]# Tensors we want to evaluatepredictions = graph.get_operation_by_name("output/predictions").outputs[0]

所以,很可能输入节点是”input_x”,输出节点是”output/predictions”。

Related Posts

关于k折交叉验证的直观问题

我在使用交叉验证检查预测能力时遇到了一些直观问题,我认…

调整numpy数组大小以使用sklearn的train_test_split函数?

我正在尝试使用sklearn中的test_train_…

如何转换二维张量和索引张量以便用于torch.nn.utils.rnn.pack_sequence

我有一组序列,格式如下: sequences = to…

模型预测值的含义是什么?

我在网上找到一个数字识别器的CNN模型并进行了训练,当…

锯齿张量作为LSTM的输入

了解锯齿张量以及如何在TensorFlow中使用它们。…

如何告诉SciKit的LinearRegression模型预测值不能小于零?

我有以下代码,尝试根据非价格基础特征来估值股票。 pr…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注