python - 导入错误 : No module named 'tensorflow.contrib.lite.python.tflite_convert'

Question

我尝试使用 tflite_convert 将我的 saved_model.pb(对象检测 API)文件转换为 .tflite，但是当我在 C:\Users\LENOVO-PC\tensorflow> 目录的 cmd 上执行此命令时 克隆 tensorflow git 的地方，

tflite_convert \ --output_file=/saved_model/maonani.tflite \ --saved_model_dir=/saved_model/saved_model

我得到一个错误提示

ImportError: No module named 'tensorflow.contrib.lite.python.tflite_convert'

完整的源日志是

C:\Users\LENOVO-PC\tensorflow>tflite_convert \ --output_file=/saved_model/maonani.tflite \ --saved_model_dir=/saved_model/saved_model
Traceback (most recent call last):
  File "c:\users\lenovo-pc\appdata\local\programs\python\python35\lib\runpy.py", line 184, in _run_module_as_main
    "__main__", mod_spec)
  File "c:\users\lenovo-pc\appdata\local\programs\python\python35\lib\runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "C:\Users\LENOVO-PC\AppData\Local\Programs\Python\Python35\Scripts\tflite_convert.exe\__main__.py", line 5, in <module>
ImportError: No module named 'tensorflow.contrib.lite.python.tflite_convert'

有没有办法在 WINDOWS 上将我的 .pb 文件转换为 .tflite？

Answer 1

您好，我的解决方案是通过以下方式使用 linux适用于 Linux 的 Windows 子系统 - see

然后从 ubunto 商店安装

然后需要 pip3 install --upgrade "tensorflow=1.7*"那么如果你尝试运行 toco，它就不会被识别。

解决方案是转到文件夹

~/.local/bin/

在那里你会找到 toco，这是一个 python 文件。

运行

python3 ~/.local/bin/toco

你会得到 toco 的“exe”。

要进行转换，您可以运行解释的命令 in https://codelabs.developers.google.com/codelabs/tensorflow-for-poets-2-tflite/#2

只需将 -graph_def_file=tf_files/retrained_graph.pb 更改为 --input_file=tf_files/retrained_graph.pb

希望这对某人有帮助

注意:如果您缺少 pip3，则需要安装它