Метод from_keras_model класса TFLiteConverter

Метод from_keras_model класса TFLiteConverter создает объект конвертера на основе уже обученной модели Keras. Первым параметром метод принимает саму Keras-модель. Вторым необязательным параметром можно передать список конкретных функций модели для конвертации. Полученный конвертер затем используется для преобразования модели в формат TensorFlow Lite с помощью метода convert.

Синтаксис

tf.lite.TFLiteConverter.from_keras_model(model, [trackable_obj])

Пример

Давайте создадим простую модель Keras и преобразуем ее в формат TensorFlow Lite:

import tensorflow as tf

tf.random.set_seed(0)

model = tf.keras.Sequential([
	tf.keras.layers.Dense(10, activation='relu', input_shape=(5,)),
	tf.keras.layers.Dense(1)
])

converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_model = converter.convert()
print(type(tflite_model))

Результат выполнения кода:

<class 'bytes'>

Пример

Давайте сохраним сконвертированную модель в файл формата .keras и проверим размер полученных данных:

import tensorflow as tf

tf.random.set_seed(0)

model = tf.keras.Sequential([
	tf.keras.layers.Dense(10, activation='relu', input_shape=(5,)),
	tf.keras.layers.Dense(1)
])

converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_model = converter.convert()

with open('model.tflite', 'wb') as f:
	f.write(tflite_model)

res = len(tflite_model)
print(res)

Результат выполнения кода:

1420

Пример

Давайте создадим модель с входным слоем и применим метод from_keras_model для дальнейшей конвертации:

import tensorflow as tf

tf.random.set_seed(0)

inputs = tf.keras.Input(shape=(5,))
x = tf.keras.layers.Dense(10, activation='relu')(inputs)
outputs = tf.keras.layers.Dense(1)(x)
model = tf.keras.Model(inputs=inputs, outputs=outputs)

converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_model = converter.convert()

print(tflite_model[:4])

Результат выполнения кода:

b'TFL3'

Смотрите также

класс TFLiteConverter,
который преобразует модели TensorFlow в формат TFLite
метод from_saved_model,
который создает конвертер из сохраненной модели
метод from_concrete_functions,
который создает конвертер из конкретных функций
метод convert,
который выполняет конвертацию модели в формат TFLite