TOCO / Tensorflow .pb в .tflite output_arrays не работает - PullRequest
Новая
       12

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

0 голосов
/ 08 июня 2018

Я относительно новичок в 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 спасибо

1 Ответ

0 голосов
/ 24 августа 2018

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

...