ValueError将Tensorflow [Toco]转换为优化格式

Question

我刚刚开始尝试Tensorflow，我想在Android上做一些实验，但是我读到Tensorflow for Poets (移动版)支持.tflite，而我下载和搜索的模型是.pb。

所以我决定用toco来转换它们，但是当我执行命令来转换时，我总是会得到这个错误。

ValueError:找到无效的张量‘输入’。

CLI命令是：

toco   --graph_def_file=tensorflow_inception_graph.pb   --output_file=optimized_graph.tflite   --input_format=TENSORFLOW_GRAPHDEF --output_format=TFLITE   --input_shape=1,300,300,3   --input_array=input   --o
utput_array=final_result   --inference_type=FLOAT   --inference_input_type=FLOAT

这有什么问题吗？

Stacktrace

File "/home/Mitro/venv/bin/toco", line 11, in <module>
    sys.exit(main())
  File "/home/Mitro/venv/lib/python3.5/site-packages/tensorflow/contrib/lite/python/tflite_convert.py", line 401, in main
    app.run(main=run_main, argv=sys.argv[:1])
  File "/home/Mitro/venv/lib/python3.5/site-packages/tensorflow/python/platform/app.py", line 125, in run
    _sys.exit(main(argv))
  File "/home/Mitro/venv/lib/python3.5/site-packages/tensorflow/contrib/lite/python/tflite_convert.py", line 397, in run_main
    _convert_model(tflite_flags)
  File "/home/Mitro/venv/lib/python3.5/site-packages/tensorflow/contrib/lite/python/tflite_convert.py", line 100, in _convert_model
    converter = _get_toco_converter(flags)
  File "/home/Mitro/venv/lib/python3.5/site-packages/tensorflow/contrib/lite/python/tflite_convert.py", line 87, in _get_toco_converter
    return converter_fn(**converter_kwargs)
  File "/home/Mitro/venv/lib/python3.5/site-packages/tensorflow/contrib/lite/python/lite.py", line 271, in from_frozen_graph
    sess.graph, input_arrays)
  File "/home/Mitro/venv/lib/python3.5/site-packages/tensorflow/contrib/lite/python/convert_saved_model.py", line 189, in get_tensors_from_tensor_names
    ",".join(invalid_tensors)))

我的谷歌参考资料

Answer 1

运行此脚本，将model_file参数更改为pb文件，作为输出，您将得到如下所示：

Tensor("import/Placeholder:0"，shape=(?，299,299，3)，dtype=float32)

Tensor("import/final_result:0"，shape=(?，5)，dtype=float32)

在这种情况下，它们是您必须通过的参数。

--Placeholder_array=占位符

--输出数组= final_result

这里是完整的信息