gpt4 book ai didi

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

转载 作者:可可西里 更新时间:2023-11-01 10:41:20 25 4
gpt4 key购买 nike

当我尝试按照 poet 的 Tensorflow 上的说明将 .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 中遇到这个错误。

请帮忙!

最佳答案

最后我在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 名称和 input_shape

关于python - Tensorflow 使用 python 在 Windows 中将 .pb 文件转换为 .lite 文件,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55120166/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com