Модель Keras HDF5 не может преобразовать Tflite - PullRequest
Новая
       91

Модель Keras HDF5 не может преобразовать Tflite

0 голосов
/ 12 июля 2020

Я запустил этот фрагмент кода в Google Colab, чтобы преобразовать мой файл Keras hdf5 в файл Tflite: 

import tensorflow as tf
keras_model = tf.keras.models.load_model("/content/best_model_11class.hdf5")
converter = tf.lite.TFLiteConverter.from_keras_model(keras_model)
tfmodel = converter.convert()
open ('model.tflite' , "wb") .write(tfmodel)

Я получаю эту ошибку при запуске кода: 

ValueError                                Traceback (most recent call last)
<ipython-input-2-0804f3b57a48> in <module>()
      2 keras_model = tf.keras.models.load_model("/content/best_model_11class.hdf5")
      3 converter = tf.lite.TFLiteConverter.from_keras_model(keras_model)
----> 4 tfmodel = converter.convert()
      5 open ('model.tflite' , "wb") .write(tfmodel)

/usr/local/lib/python3.6/dist-packages/tensorflow/lite/python/lite.py in convert(self)
    481               "None is only supported in the 1st dimension. Tensor '{0}' has "
    482               "invalid shape '{1}'.".format(
--> 483                   _get_tensor_name(tensor), shape_list))
    484         elif shape_list and shape_list[0] is None:
    485           # Set the batch size to 1 if undefined.

ValueError: None is only supported in the 1st dimension. Tensor 'input_1' has invalid shape '[None, None, None, 3]'.

Может кто подскажет, как это исправить?

1 Ответ

0 голосов
/ 04 августа 2020

Используйте этот блокнот Google Colab для конвертации. Загрузите файл .hdf5, и он преобразует его в файл .tflite.

Ссылка на записную книжку https://colab.research.google.com/drive/1E0dqzDYszsG2BfyzVWUOdKLElqvMn4oo?usp=sharing

После выполнения последнего кода go в файлы ( крайняя левая сторона) там вы найдете модель .tflite

...