Ошибка при преобразовании модели тензорного потока в тензорный поток lite (.h5 в .tflite) = ValueError: неверные данные маршала (код неизвестного типа)

Question

Я использую предварительно обученную модель на основе кераса и тензорного потока с архитектурой yolov2 для обнаружения выбоин, и у меня возникает ошибка при преобразовании моей модели тензорного потока в тензорный поток lite

from tensorflow.contrib import lite

converter = lite.TFLiteConverter.from_keras_model_file( 'model.h5' )
model = converter.convert()
file = open( 'model.tflite' , 'wb' )
file.write( model )

Answer 1

Я бы предложил пойти по более длинному, но проверенному пути.

1) Конвертируйте вашу модель напрямую в .pb формат замороженного графика или конвертируйте из h5 в .pb, используя Как экспортировать Keras .h5 в тензор потока .pb?

2) Преобразовать .pb в tflite, как описано мной здесь: https://medium.com/@prashantdandriyal7 / дар knet -тензор-поток-тензор-поток-b783d55b106a