У меня есть мой сохраненный_модель.pb вместе с папкой переменных (но она пустая).
Модель Mask RCNN и имеет только один класс.
Я бы хотел конвертировать .pb в веб-формат TensorFlow.js. Я сталкиваюсь с проблемами, пытаясь угадать, какие выходные_имя_имен передать в команду tensorflowjs_converter
.
Я использовал этот скрипт, но он печатает много имен узлов (много)!
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)
Я также использовал тензорную доску, чтобы получить output_node_names, но ничего не отображается.