Как найти узлы ввода и вывода замороженной модели

Я хочу использовать сценарий tenorflow optimize_for_inference.py на замороженной модели из зоопарка моделей: ssd_mobilenet_v1_coco.

Как мне найти / определить имена входного и выходного имени модели?

Вот ссылка на график, созданный тензорной таблицей

Этот вопрос может помочь: Учитывая граф тензорной модели потока, как найти имена входных и выходных узлов (для меня это не так)


person gustavz    schedule 11.01.2018    source источник


Ответы (1)


Мы добавили некоторую документацию по этому процессу, описанную здесь:

https://www.tensorflow.org/mobile/prepare_models

Если вы поищете summarize_graph, вы увидите пример того, как определить правильные узлы ввода и вывода.

person Pete Warden    schedule 05.02.2018
comment
Последняя документация находится здесь: tensorflow.org/lite/guide/ Если они снова переместят его, они предлагают использовать: bazel build tensorflow/tools/graph_transforms:summarize_graph bazel-bin/tensorflow/tools/graph_transforms/summarize_graph --in_graph=tensorflow_inception_graph.pb или python import_pb_to_tensorboard.py --model_dir <model path> --log_dir <log dir path> - person jb455; 29.04.2019