tensorflow - 如何将 tflite 模型转换为 Tensorflow 中的卡住图 (.pb)？

Question

我想将整数量化 tflite 模型转换为 Tensorflow 中的卡住图 (.pb)。我通读并尝试了 StackOverflow 上的许多解决方案，但没有一个有效。具体来说，toco 不起作用(output_format 不能是 TENSORFLOW_GRAPHDEF)。

我的最终目标是通过tf2onnx得到一个量化的ONNX模型，但是tf2onnx不支持tflite作为输入(只支持saved_model、checkpoint和graph_def)。然而，在使用 TFLiteConverter 量化训练模型后，它只返回一个 tflite 文件。这就是问题所在。

理想的流程本质上是这样的:float32 中的 tf 模型 -> int8 中的 tflite 模型 -> graph_def -> onnx 模型。我卡在了第二个箭头处。

Answer 1

将 tflite 模型转换为 .pb 的功能在 Tensorflow r1.9 版本后被删除。尝试将你的 TF 版本降级到 1.9，然后像这样

bazel run --config=opt \
  //tensorflow/contrib/lite/toco:toco -- \
  --input_file=/tmp/foo.tflite \
  --output_file=/tmp/foo.pb \
  --input_format=TFLITE \
  --output_format=TENSORFLOW_GRAPHDEF \
  --input_shape=1,128,128,3 \
  --input_array=input \
  --output_array=MobilenetV1/Predictions/Reshape_1

Here是来源。