tensorflow Lite之转换为.tflite文件遇到的问题
已有.pb,.ckpt文件,想要生成.tflite文件,然后把模型移到手机上。在tensorflow lite官网上看到转换为.tflite命令如下:
tflite_convert \
--output_file=/tmp/mobilenet_v1_1.0_224.tflite \
--graph_def_file=/tmp/mobilenet_v1_0.50_128/frozen_graph.pb \
--input_arrays=input \
--output_arrays=MobilenetV1/Predictions/Reshape_1
求问大神们报错如下:怎么解决呢?(大概是输入的时候和队列有关所以出了问题吗?)不知道怎么解决,急求解答
Traceback (most recent call last):
File "/home/dy/.local/bin/tflite_convert", line 11, in <module>
sys.exit(main())
File "/home/dy/.local/lib/python3.5/site-packages/tensorflow/contrib/lite/python/tflite_convert.py", line 412, in main
app.run(main=run_main, argv=sys.argv[:1])
File "/home/dy/.local/lib/python3.5/site-packages/tensorflow/python/platform/app.py", line 125, in run
_sys.exit(main(argv))
File "/home/dy/.local/lib/python3.5/site-packages/tensorflow/contrib/lite/python/tflite_convert.py", line 408, in run_main
_convert_model(tflite_flags)
File "/home/dy/.local/lib/python3.5/site-packages/tensorflow/contrib/lite/python/tflite_convert.py", line 162, in _convert_model
output_data = converter.convert()
File "/home/dy/.local/lib/python3.5/site-packages/tensorflow/contrib/lite/python/lite.py", line 410, in convert
elif shape[0] is None:
IndexError: list index out of range