Невозможно преобразовать файл .h5 в модель tflite

Question

Я обучил модель и сохранил ее как файл h5. Поскольку я хочу использовать это в своем приложении android, я хочу преобразовать его в tflite на Colab. Это мой код:

import tensorflow as tf
model = tf.keras.models.load_model('Final_model.h5')
converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

Я получаю ошибку:

AttributeError: type object 'TFLiteConverter' has no attribute 'from_keras_model'

Как мне решить эту проблему?

Answer 1

Вы используете colab's по умолчанию tensorflow, то есть 1.15.0.

Вы можете скачать новейшую версию, выполнив из ячейки colab:

!pip install --upgrade pip && pip install tensorflow.

Возможно, вам придется заранее удалить tensorflow версию 1.15.0 с помощью этой команды:

!pip uninstall tensorflow (or tensorflow-gpu).

После этого вы сможете нормально запускать код tf2.x. .

Примечание: Скоро версия TF2.0 станет версией по умолчанию, и этот обходной путь временный.

Answer 2

Вы используете Tensorflow 1.x, который в этом контексте имеет немного другой API . То, что вы должны использовать это from_keras_model_file, то есть:

converter = tf.lite.TFLiteConverter.from_keras_model_file('Final_model.h5')