Модель tf.keras HDF5 и модель Keras HDF5 - PullRequest
Новая
       11

Модель tf.keras HDF5 и модель Keras HDF5

1 голос
/ 03 мая 2019

Я хочу преобразовать модель Keras в модель Tensorflow Lite.Когда я изучал документацию, было указано, что мы можем использовать модели tf.keras HDF5 в качестве входных данных.Означает ли это, что я могу использовать свою сохраненную модель HDF5 Keras в качестве входных данных или модели tf.keras HDF5 и модели Keras HDF5 - это разные вещи?

Документация: https://www.tensorflow.org/lite/convert

Редактировать: я могпреобразовать мою модель Keras в модель Tensorflow Lite с использованием этого API, но я еще не тестировал ее.Мой код: 

converter = tf.lite.TFLiteConverter.from_keras_model_file(path + 'plant- 
recognition-model.h5')
tflite_model = converter.convert()

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

1 Ответ

1 голос
/ 04 мая 2019

Модель tf.keras HDF5 и модели Keras HDF5 - это не разные вещи, за исключением неизбежной синхронности обновления версии программного обеспечения. Это то, что официальные документы говорят :

tf.keras - это реализация спецификации Keras API в TensorFlow. Это высокоуровневый API для построения и обучения моделей, включающий первоклассную поддержку специфичной для TensorFlow функциональности

Если конвертер сможет конвертировать модель keras в tf.lite, он даст те же результаты. Но функциональность tf.lite более ограничена, чем tf.keras. Если вам не хватает этого набора функций, вы все равно можете работать с tenorflow и пользоваться его другими преимуществами.

Может быть, это не займет много времени, прежде чем ваши модели могут работать на смартфоне.

...