如何找到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近邻算法时,有没有办法获取被使用的“邻居”?

我想找到一种方法来确定在我的knn算法中实际使用了哪些…

Theano在Google Colab上无法启用GPU支持

我在尝试使用Theano库训练一个模型。由于我的电脑内…

准确性评分似乎有误

这里是代码: from sklearn.metrics…

Keras Functional API: “错误检查输入时:期望input_1具有4个维度,但得到形状为(X, Y)的数组”

我在尝试使用Keras的fit_generator来训…

如何使用sklearn.datasets.make_classification在指定范围内生成合成数据?

我想为分类问题创建合成数据。我使用了sklearn.d…

如何处理预测时不在训练集中的标签

已关闭。 此问题与编程或软件开发无关。目前不接受回答。…

发表回复

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