重要错误:没有名为'tensorflow.contrib.lite.python.tflite_转换'

2024-03-29 12:54:05 发布

您现在位置:Python中文网/ 问答频道 /正文

我尝试用tflite_convert转换我保存的_型号.pb(对象检测API)文件到.tflite,但当我在克隆tensorflow git的C:\Users\LENOVO-PC\tensorflow>目录上的cmd上执行此命令时

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

我有个错误说

^{pr2}$

完整的源日志是

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文件?在


Tags: 文件runinpyconvertmodeltensorflowline
3条回答

嗨,我的解决方案是按照以下方式使用linux Windows Linux子系统-see

然后从仓库Ubento安装

然后需要pip3安装升级“tensorflow=1.7*” 如果你试图运行toco,它将不会被识别。在

解决方案是转到文件夹

~/.local/bin/

在那里你可以找到toco,这是一个python文件。在

python3 ~/.local/bin/toco

你会得到toco的“exe”。在

要转换,可以运行解释的命令inhttps://codelabs.developers.google.com/codelabs/tensorflow-for-poets-2-tflite/#2

只需更改-graph_def_file=tf_files/rettrained_图形.pb输入_file=tf_files/rettrained_图形.pb在

希望有人能帮上忙

注: 如果缺少pip3,则需要安装它

根据这个线索:Tensorflow discussions

问题是,目前为止,windows不支持该模块。你可以跟踪这个线程,看看是否有关于它的更新。在

附言:有些人声称git克隆和bazel构建帮助解决了这个问题,而不是pip安装,您也可以尝试一下,但对这是否有效表示怀疑。在

我遵循这个网站上的说明:

https://codelabs.developers.google.com/codelabs/tensorflow-for-poets-2-tflite/#2

但是,tflearn_convert似乎不再支持Windows了。所以我决定 在Windows上使用Ubuntu。在创建了一个安装tensorflow的虚拟环境之后,我通过在终端中输入toco来检查“toco”。这是使用toco的指令。在

usage: /home/hieu/venv/bin/toco

Flags:

^{pr2}$

还有更多。。。在

之后,我使用以下命令转换文件:

 toco  input_file="tf_files/retrained_graph.pb"  output_file="tf_files/optimized_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"

然后优化_图形.lite应在tf\u文件中找到

相关问题 更多 >