将Frozen.pb文件转换为tflite格式时出错

问题描述 投票:0回答:1
tflite_convert   --output_file=./graph.tflite   --graph_def_file=output_graph_frozen.pb   --input_arrays=IteratorV2 --output_arrays=linear/head/predictions/probabilities

追踪(最近通话):在第11行的文件“ / usr / local / bin / tflite_convert”sys.exit(main())主目录中的文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/lite/python/tflite_convert.py”,第503行app.run(main = run_main,argv = sys.argv [:1])运行中的文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/python/platform/app.py”,第40行_run(main = main,argv = argv,flags_parser = _parse_flags_tolerate_undef)运行中的文件“ /usr/local/lib/python3.6/dist-packages/absl/app.py”,第299行_run_main(main,args)_run_main中的文件“ /usr/local/lib/python3.6/dist-packages/absl/app.py”,第250行sys.exit(main(argv))在run_main的第499行中,文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/lite/python/tflite_convert.py”_convert_tf1_model(tflite_flags)_convert_tf1_model中的第193行的“ /usr/local/lib/python3.6/dist-packages/tensorflow/lite/python/tflite_convert.py”文件output_data = converter.convert()在转换文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/lite/python/lite.py”中,第898行** converter_kwargs)toco_convert_impl中的第401行的文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/lite/python/convert.py”输入张量,输出张量,* args,** kwargs)在build_toco_convert_protos中的文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/lite/python/convert.py”,第304行input_tensor.dtype)文件“ /usr/local/lib/python3.6/dist-packages/tensorflow/lite/python/util.py”,第58行,在convert_dtype_to_tflite_type中引发ValueError(“不支持的tf.dtype {0}”。format(tf_dtype))ValueError:不支持的tf.dtype

tensorflow tensorflow-lite
1个回答
0
投票
设置“ input_arrays”和“ output_arrays”的目的是什么?

在最简单的情况下,您可以做

converter = tf.lite.TFLiteConverter.from_saved_model(export_dir) tflite_model = converter.convert()

https://www.tensorflow.org/lite/convert/python_api所述。
© www.soinside.com 2019 - 2024. All rights reserved.