在Windows上通过命令行将Tensorflow模型转换为Tflite模型时遇到麻烦

Question

在Windows上通过命令行将张量流模型转换为tflite模型时遇到麻烦。

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：未命名模块   'tensorflow.contrib.lite.python.tflite_convert'

问题是在Windows上使用tflite_convert造成的。

Answer 1

当我遇到同样的问题时。我已经使用tensorFlow晚安版解决了它。 这是我使用的命令：

pip install tf-nightly

这是在navigator env中安装它的正确方法。 tf-nighty

我希望这会有所帮助。