Path: blob/master/site/zh-cn/lite/r1/convert/index.md
25118 views
TensorFlow Lite converter
TensorFlow Lite converter是用于将TensorFlow模型转化为优化过的FlatBuffer格式,以便让TensorFlow Lite解释器调用。
注意:此页面包含TensorFlow 1.x的converter API文档,TensorFlow 2.0的API请点击此链接
FlatBuffers
FlatBuffers是一个高效的开源跨平台序列化库。它类似于protocol buffers,区别在于FlatBuffers在访问数据之前不需要对其次要表达进行解析/解压,从而避免对每个对象进行内存分配。FlatBuffers的代码占用空间比protocol buffers小一个数量级。
从模型培训到设备部署
TensorFlow Lite converter可以从TensorFlow模型中生成TensorFlow Lite FlatBuffers文件(.tflite)。
converter支持以下输入格式:
变量固定为常数(Frozen)的
GraphDef:由freeze_graph.py生成的模型tf.kerasHDF5模型任何从
tf.Session获取的模型(仅限Python API)
然后,将TensorFlow Lite FlatBuffer文件部署到客户端设备,TensorFlow Lite 解释器会使用压缩模型在设备上进行推断(inference)。该会话过程如下图所示:
选项
TensorFlow Lite Converter 可以通过以下两种方式使用: