怎么把.pb转换成.tflite？

Question

我已经使用Pytorch创建了一个对象检测模型，然后从.pth转换为.onnx，然后是.pb，但是现在我需要将其转换为{{1 }} for android app！怎么做？这是我的第一次。

.tflite

用于二进制分类。

我已经从pytorch完成了，但是我发现要看的一切都是来自keras或Tensorflow ...

这是我用来将其从input_arrays = [64, 3, 224, 224] output_arrays = ? 转换为.pb的代码

.tflie

converter = lite.TFLiteConverter.from_frozen_graph(
  model/model.pb , input_arrays, output arrays )
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

我认为这与输入数组和输出数组有关，但不确定。 !tflite_convert \ --output_file= model/model.tflite \ --graph_def_file= model/model.pb \ --input_arrays= input_arrays \ -- output_arrays= output_arrays 是否应该存储graph_def_file？

Answer 1

使用以下代码时，无需指定输入和输出数组：

import tensorflow as tf

converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

尝试一下。