Как была преобразована модель tflite "conv_actions_tflite", предоставляемая демо-программой распознавания речевых команд? - PullRequest
Новая
       23

Как была преобразована модель tflite "conv_actions_tflite", предоставляемая демо-программой распознавания речевых команд?

0 голосов
/ 23 октября 2019

Я использовал lite Converter для преобразования моей модели pb format в tflite format в терминале, но она не работала хорошо.

Но когда я использовал tflite model, предоставленный speech command android demo, онработает довольно хорошоИтак, я хочу знать, как эта модель была преобразована?

https://github.com/tensorflow/docs/blob/master/site/en/r1/tutorials/sequences/audio_recognition.md

Используя приведенную выше ссылку, я обучил модель с помощью следующей команды

(base) unizen@admin:~/tensorflow/tensorflow/examples/speech_commands$ python train.py

Когда модель сохраняется после тренировки, я создал замороженную модель, используя следующий код: 

(base) unizen@admin:~/tensorflow/tensorflow/examples/speech_commands$ python freeze.py \
--start_checkpoint=/tmp/speech_commands_train/conv.ckpt-18000 \
--output_file=/tmp/my_frozen_graph.pb

Но когда я попытался преобразовать .pb format в tflite format

(base) unizen@admin:~/tensorflow/tensorflow/examples/speech_commands$  tflite_convert \
--saved_model_dir  /home/unizen/Downloads/my_frozen_graph.pb \
--input_format TENSORFLOW_GRAPHDEF \
--input_arrays decoded_sample_data \
--input_shapes 16000,1 \
--output_arrays labels_softmax \
--output_format TFLITE \
--output_file /tmp/sprc.tflite \
--allow_custom_ops

ошибка 

(base) unizen@admin:~/tensorflow/tensorflow/examples/speech_commands$ python usage: tflite_convert [-h] --output_file OUTPUT_FILE
                      (--saved_model_dir SAVED_MODEL_DIR | --keras_model_file KERAS_MODEL_FILE)
tflite_convert: error: one of the arguments --saved_model_dir --keras_model_file is required.

пожалуйста, предоставьте решение для преобразования замороженной модели в модель tflite

...