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

1

У меня проблема с преобразованием модели тензорного потока в 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.

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

Теги:
tensorflow
tensorflow-lite

2 ответа

0

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

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

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

Ещё вопросы

Сообщество Overcoder
Наверх
Меню