[英]How to convert from .pb to .tflite?
我一直在使用创建的对象检测模型Pytorch
,然后从转换.pth
到.onnx
然后.pb
,但现在我需要把它转换成.tflite
的Android应用程序! 怎么做? 这是我的第一次。
input_arrays = [64, 3, 224, 224]
output_arrays = ?
用于二进制分类。
我是从pytorch完成的,但我发现的所有内容都来自keras或Tensorflow ......
这是我用来将它从.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
应该存储model.pb
吗?
使用以下代码时无需指定输入和输出数组:
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)
试试吧。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.