tensorflow Lite之转换为.tflite文件遇到的问题

dydxdt 2019-01-07 09:22:36
已有.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
...全文
534 1 打赏 收藏 转发到动态 举报
写回复
用AI写文章
1 条回复
切换为时间正序
请发表友善的回复…
发表回复
babytiger 2019-04-17
  • 打赏
  • 举报
回复
我在py3.6,linux ubuntu16.04可以,换成windows10就不行 了,建议换系统

80,351

社区成员

发帖
与我相关
我的任务
社区描述
移动平台 Android
androidandroid-studioandroidx 技术论坛(原bbs)
社区管理员
  • Android
  • yechaoa
  • 失落夏天
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧