У меня проблема с преобразованием модели тензорного потока в tflite. У меня есть изученная модель, основанная на Tensorflow Object Detection. Я хотел бы использовать код преобразования из конвертера TFlite.
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.
Спасибо за ответы
Если вы загрузите модель отсюда, то появится файл mobilenet_v1_0.5_128_info.txt, в котором указано имя входа и выхода
Model: mobilenet_v1_0.5_128
Input: input
Output: MobilenetV1/Predictions/Reshape_1
У нас есть скрипт в API обнаружения объектов для получения Flatbuffer.