Как получить имя ввода / вывода для создания файла .pb

Question

Я обучил модель для API обнаружения объектов Tensorflow с моим набором данных. После сохранения модели я получил три файла:

• model.ckpt-10000.data-00000-оф-00001
• model.ckpt-10000.meta
• model.ckpt-10000.index

Из этих файлов я хочу создать файл .pb. Я погуглил и понял, что могу использовать «freeze_graph.py», который содержится в проекте Tensorflow. Но чтобы использовать freeze_graph.py, мне нужны имена ввода / вывода.

Некоторые люди предлагали использовать «инструмент summaze_graph», но для этого мне нужно предоставить файл .pb в аргументах. Мне нужны имена вывода / ввода для создания файла .pb, поэтому я не могу использовать «инструмент суммирования_графа», так как у меня еще нет файла .pb.

Я видел этот вопрос: Как вы получаете имя выходных узлов тензорного потока в модели Keras?
Но он просто дал мне тысячи возможных имен узлов вывода / ввода.

Итак, мой вопрос:
Как получить узел ввода / вывода без файла .pb?

Answer 1

Я обнаружил, что есть инструмент для замораживания графика для API обнаружения объектов, который можно использовать без имени ввода / вывода. Смотрите команду здесь . Это так:

# From tensorflow/models/research/
python object_detection/export_inference_graph.py \
    --input_type image_tensor \
    --pipeline_config_path ${PIPELINE_CONFIG_PATH} \
    --trained_checkpoint_prefix ${TRAIN_PATH} \
    --output_directory output_inference_graph.pb

Затем я получил файл .pb из этих трех файлов: model.ckpt-10000.data-00000-of-00001, model.ckpt-10000.meta, model.ckpt-10000.index.

Или, если у вас уже есть файл .pb, вы можете использовать это: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/graph_transforms/README.md#inspecting-graphs