如何获取tensorflowjs_converter命令的output_node_names?

时间:2019-05-07 13:19:49

标签: tensorflow tensorflowjs-converter

我有我的saved_model.pb和变量文件夹(但它是空的)。 该模型是Mask RCNN,仅具有一个类。

我想将.pb转换为TensorFlow.js Web格式。我在猜测要在tensorflowjs_converter命令中传递哪个output_node_names时遇到麻烦。 我使用了此脚本,但它会打印很多节点的名称(很多)!

import tensorflow as tf

modelName = 'path_to_checkpoints'

tf.reset_default_graph()
with tf.Session() as sess:
    saver = tf.train.import_meta_graph(modelName)
    graph_def = tf.get_default_graph().as_graph_def()
    node_list=[n.name for n in graph_def.node]


print(node_list)

我还使用了tensorboard来获取output_node_names,但是什么也没显示。

0 个答案:

没有答案