Question

我使用mobilenet_1.0_224训练了数据集以生成模型文件（.pb扩展名）。我想使用TFLite开发Android图像识别应用程序。但是，当我尝试使用toco将模型优化为.lite扩展时，已生成此错误。我不使用bazel-build，因为无论我如何安装它，bazel都无法识别我的输入文件，即graph.pb。有没有办法解决这个问题？

优化使用的脚本：

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,${IMAGE_SIZE},${IMAGE_SIZE},3 \
>   --input_array=input \
>   --output_array=final_result \
>   --inference_type=FLOAT \
>   --input_type=FLOAT

生成错误：

Traceback (most recent call last):
  File "c:\python36\lib\runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "c:\python36\lib\runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "C:\Python36\Scripts\toco.exe\__main__.py", line 5, in <module>
ModuleNotFoundError: No module named 'tensorflow.contrib.lite.toco.python'

对于tflite Android，Tensorflow toco无法将graph.pb转换为graph.lite

0 个答案: