我已经使用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
?
答案 0 :(得分: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)
尝试一下。