Столкновение с проблемой при преобразовании модели тензорного потока в tflite через командную строку в Windows

Question

Столкновение с проблемой при преобразовании модели тензорного потока в модель tflite через командную строку в Windows.

IMAGE_SIZE = 224

tflite_convert \
  --graph_def_file=tf_files/retrained_graph.pb \
  --output_file=tf_files/optimized_graph.lite \
  --input_format=TENSORFLOW_GRAPHDEF \
  --output_format=TFLITE \
  --input_shape=1,${IMAGE_SIZE},${IMAGE_SIZE},3 \
  --input_array=input \
  --output_array=final_result \
  --inference_type=FLOAT \
  --input_data_type=FLOAT

всегда выдает ошибку

ModuleNotFoundError: нет модуля с именем 'tenorflow.contrib.lite.python.tflite_convert'

проблема с tflite_convert в windows.

Answer 1

Когда у меня возникла такая же проблема.Я решил это, используя тензорную версию тенорФлоу.Вот команда, которую я использовал:

pip install tf-nightly

И вот правильный способ ее установки в env навигатора. tf-nighty

Надеюсь, это будет полезно.