python - Tensorflow [Toco] 将模型转换为优化格式导致 ValueError
问题描述
我刚开始尝试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
那有什么问题?
堆栈跟踪
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)))
解决方案
推荐阅读
- ios - 如何从 Array 中播放特定的声音?
- java - 异常“java.lang.IllegalStateException:在 JobRepository 中检测到现有事务。” 在将 Spring Batch 从 2.1.1 升级到 4.1.1 版本之后
- javascript - gcp 上的 html 中的 js(本地测试)不起作用
- jquery - 解析 JSON Google Place 数据时遇到问题
- javascript - Smartlook 未正确保存用户同意
- spring - 使用 Spring Kafka 进行 Kafka 主题排序
- java - 如何使 Spring Boot 在其重新打包的 WAR 中包含清单文件?
- javascript - 不活动后自动注销 Drupal 8
- javascript - 在 Cucumber 中,虽然调用了回调,但回调函数却超时了。已经增加了 DefaultTimeout 但仍然错误
- angular - 如何修复 AssertionError [ERR_ASSERTION]: false == true?