我有我的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,但是什么也没显示。