TOCO/Tensorflow .pb к.tflite output_arrays не работает

Question

Я относительно новичок в Tensorflow, и я создал специально обученную модель изображения, которая является моделью переобученной mobilenetv1. В настоящее время у меня есть график.pb и файл label.txt.

Я пытаюсь использовать TOCO, чтобы преобразовать мой граф.pb в граф.tflite, чтобы я мог использовать его с Firebase, и у меня постоянно появляется ошибка при попытке запустить TOCO. Код, который я запускаю в терминале:

./toco \
--input_file=/***/frozen_graph.pb \
--input_format=TENSORFLOW_GRAPHDEF \
--output_format=TFLITE \
--output_file=/***/newGraph.tflite \
--inference_type=FLOAT \
--input_arrays=input \
--input_shapes=1,224,224,3 \

Я попытался указать --output_arrays флаг, но я не знаю, что для него установить. Ошибка, которую я получаю в терминале:

F tensorflow/contrib/lite/toco/tooling_util.cc:1480] Check failed: model->flags.output_arrays_size() > 0 (0 vs. 0)This model does not define output arrays, so a --output_arrays flag must be given on the command-line.
Abort trap: 6

Я потратил столько часов и часов, пытаясь понять, что установить --output_arrays равно, и я застрял. Я использую MacOS 10.13.3. Пожалуйста, помогите meeeeeeeee спасибо

Answer 1

--output_arrays является обязательным флагом в TOCO. Большинство моделей MobileNetV1 имеют выходной массив MobilenetV1/Predictions/Reshape_1Однако это может варьироваться в зависимости от вашей модели. Лучший способ просмотреть выходной массив - использовать TensorBoard для визуализации вашего графика.