Конвертировать модель keras из файла pb в файл tflite

Question

Я пытаюсь преобразовать обученную модель Keras из файла .pb в файл .tflite:

tflite_model = tf.contrib.lite.toco_convert(frozen_graph_def, input_tensors, output_tensors)

Однако, когда я запускаю этот код, появляется ошибка, которая говорит:

AttributeError: модуль 'tenorflow.contrib.lite' не имеет атрибута 'Toco_convert'

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

Answer 1

Используете ли вы последнюю версию tenorflow? это также можно решить при установке tf-nightly. Или составление тензорного потока из источника с помощью базеля. Вы также можете использовать вместо: tf.contrib.lite.TocoConverter.from_frozen_graph(final_model_path, input_arrays, output_arrays)

Answer 2

Если вы работаете в Windows, оно не будет работать, поскольку библиотека Toco не поддерживает Windows. Так что лучше переключиться на Linux и повторить попытку.

Answer 3

Пожалуйста, предоставьте график тензорного потока, который вы пытаетесь преобразовать, чтобы воспроизвести точную ошибку. Это также поможет подать проблему на GitHub https://github.com/tensorflow/tensorflow/issues