Как получить output_node_names для команды tenorflowjs_converter? - PullRequest
0 голосов
/ 07 мая 2019

У меня есть мой сохраненный_модель.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, но ничего не отображается.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...