Tensorflow Преобразование файла .pb в файл .lite в Windows с использованием Python

Question

Когда я пытаюсь преобразовать файл .pb в файл .tflite, следуя инструкции по Tensorflow для поэта: https://codelabs.developers.google.com/codelabs/tensorflow-for-poets-2-tflite/#2

Я получаю No module named 'tensorflow.contrib.lite.python.tflite_convert'

Версия My Tensorflow: 1.10.0

Я попытался запустить две команды, но это приводит к той же ошибке:

Команда 1: tflite_convert --graph_def_file=rounded_graph.pb --output_file=new_graph.lite --input_format=TENSORFLOW_GRAPHDEF --output_format=TFLITE --input_shape=1,224,224,3 --input_array=input --output_array=final_result --inference_type=FLOAT --input_data_type=FLOAT

Команда 2: toco --input_file="rounded_graph.pb" --output_file="new_graph.lite" --input_format="TENSORFLOW_GRAPHDEF" --output_format="TFLITE" --input_shape="1,224,224,3" --input_array="input" --output_array="final_result" --inference_type="FLOAT" --input_data_type="FLOAT"

Я получаю эту ошибку в windows.

Пожалуйста, помогите!

Answer 1

Наконец-то я решил эту проблему на windows:

1. Открыть CMD
2. Прогон: pip install --upgrade pip
3. Затем запустите: pip install tensorflow --upgrade

Моя новая версия Tensorflow : 1.13.0

Тогда по этой ссылке: https://codelabs.developers.google.com/codelabs/tensorflow-for-poets-2-tflite/#2

Попробуйте выполнить следующую команду : tflite_convert --graph_def_file=rounded_graph.pb --output_file=new_graph.lite --input_format=TENSORFLOW_GRAPHDEF --output_format=TFLITE --input_shape=1,224,224,3 --input_array=input --output_array=final_result --inference_type=FLOAT --input_data_type=FLOAT

Если вы получили ошибку в input , попробуйте изменить имена input и input_shape