Конвертировать модель Tensorflow в Tensorflow Lite

Question

У меня проблема с преобразованием модели тензорного потока в tflite. У меня есть изученная модель, основанная на обнаружении объекта Tensorflow Я хотел бы использовать конверсионный код из TFlite converter

curl https://storage.googleapis.com/download.tensorflow.org/models/mobilenet_v1_0.50_128_frozen.tgz | tar xzv -C /tmp

tflite_convert \
 --output_file=/tmp/foo.tflite \
 --graph_def_file=/tmp/mobilenet_v1_0.50_128/frozen_graph.pb \
 --input_arrays=input \
 --output_arrays=MobilenetV1/Predictions/Reshape_1

Я не знаю, где взять значения input_arrays и output_arrays.

Спасибо за ответы

Answer 1

Если вы загрузите модель из здесь , то будет файл mobilenet_v1_0.5_128_info.txt, в котором указано имя входа и выхода

Model: mobilenet_v1_0.5_128
Input: input
Output: MobilenetV1/Predictions/Reshape_1

Answer 2

У нас есть скрипт в API обнаружения объектов для получения Flatbuffer.