Получить имя входного и выходного узла из файлов тензорного потока .pb и .ckpt - PullRequest
Новая
       6

Получить имя входного и выходного узла из файлов тензорного потока .pb и .ckpt

0 голосов
/ 14 января 2020

Я хочу преобразовать предварительно обученную модель: mobilenetv2_dm05_coco_voc_trainaug в модель TFlite. У меня есть 3 файла:

  • frozen_inference_graph.pb
  • model.ckpt.data-00000-of-00001
  • model.ckpt.index

Чтобы преобразовать эту модель, мне нужен файл .pb, который у меня есть, и имена узлов ввода и вывода, которых у меня нет. Как я могу получить эти имена? Мой код должен быть таким: 

import tensorflow as tf

graph_def_file =  frozen_inference_graph.pb # the .pb file
input_arrays = ["Input"] # Input node
output_arrays = ["Output"] # Output node

converter = tf.contrib.lite.TocoConverter.from_frozen_graph(graph_def_file, input_arrays, output_arrays)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)
...