Tensorflow 使用 python 在 windows 中将 .pb 文件转换为 .lite 文件

Question

当我尝试按照 Tensorflow for 诗人的说明将 .pb 文件转换为 .tflite 文件时：https://codelabs.developers.google.com/codelabs/tensorflow-for-poets-2-tflite/#2

我收到No module named 'tensorflow.contrib.lite.python.tflite_convert'

我的 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_shape