此页面由 Cloud Translation API 翻译。
Switch to English

tf.lite.TFLiteConverter

TensorFlow 1版本 在GitHub上查看源代码

将TensorFlow模型转换为TensorFlow Lite模型。

用于笔记本

在指南中使用在教程中使用

用法示例:

 # Converting a SavedModel to a TensorFlow Lite model.
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
tflite_model = converter.convert()

# Converting a tf.Keras model to a TensorFlow Lite model.
converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_model = converter.convert()

# Converting ConcreteFunctions to a TensorFlow Lite model.
converter = tf.lite.TFLiteConverter.from_concrete_functions([func])
tflite_model = converter.convert()
 

funcs TensorFlow具体功能列表。该列表不应包含重复的元素。
trackable_obj tf.AutoTrackable对象关联funcs 。需要维护对该对象的引用,以免变量被垃圾回收,因为函数对变量的引用较弱。仅当用户未维护tf.AutoTrackable对象(例如from_saved_model )时才需from_saved_model

allow_custom_ops 布尔值,指示是否允许自定义操作。如果为False,则任何未知操作都是错误。如果为True,则会为任何未知的操作创建自定义操作。开发人员需要使用自定义解析器将它们提供给TensorFlow Lite运行时。 (默认为False)
optimizations 实验性标志,可能随时更改。转换模型时要应用的优化列表。例如[Optimize.DEFAULT]
representative_dataset 可以用来为模型生成输入和输出样本的代表性数据集。转换器可以使用数据集评估不同的优化。请注意,这是一个可选属性,但是如果INT8是目标操作中唯一支持内置操作的操作,则这是必需的。
target_spec 实验性标志,可能随时更改。目标设备的规格。
inference_input_type 输入图层的数据类型。请注意,整数类型(tf.int8和tf.uint8)当前仅支持训练后整数量化。 (默认tf.float32,必须位于{tf.float32,tf.int8,tf.uint8}中)
inference_output_type 输出层的数据类型。请注意,整数类型(tf.int8和tf.uint8)当前仅支持训练后整数量化。 (默认tf.float32,必须位于{tf.float32,tf.int8,tf.uint8}中)
experimental_new_converter 实验性标志,可能随时更改。启用基于MLIR的转换,而不是TOCO转换。 (默认为True)

方法

convert

查看资料

根据实例变量转换TensorFlow GraphDef。

退货
以序列化格式转换的数据。

加薪
ValueError 没有指定具体功能。指定了多个具体功能。未指定输入形状。无效的量化参数。

from_concrete_functions

查看资料

从ConcreteFunctions创建一个TFLiteConverter对象。

精氨酸
funcs TensorFlow具体功能列表。该列表不应包含重复的元素。当前,转换器只能转换单个ConcreteFunction。正在开发多种功能。

退货
TFLiteConverter对象。

加薪
输入类型无效。

from_keras_model

查看资料

从Keras模型创建TFLiteConverter对象。

精氨酸
model tf.Keras.Model

退货
TFLiteConverter对象。

from_saved_model

查看资料

从SavedModel目录创建一个TFLiteConverter对象。

精氨酸
saved_model_dir 要转换的SavedModel目录。
signature_keys 标识包含输入和输出的SignatureDef的键的列表。元素不应重复。默认情况下,使用MetaGraphdef的signatures属性。 (默认的saved_model.signatures)
tags 一组标签,用于标识要分析的SavedModel中的MetaGraphDef。标签集中的所有标签都必须存在。 (默认设置(SERVING))

退货
TFLiteConverter对象。

加薪
无效的签名密钥。