Я пытаюсь преобразовать обученную модель Keras из файла .pb в файл .tflite:
tflite_model = tf.contrib.lite.toco_convert(frozen_graph_def, input_tensors, output_tensors)
Однако когда я запускаю этот код, появляется сообщение об ошибке:
AttributeError: модуль tenorflow.contrib.lite не имеет атрибута toco_convert
Как решить эту проблему?
Если вы используете Windows, это не сработает, поскольку библиотека toco не поддерживает окна. Так что лучше переключиться на Linux и попробовать еще раз.
Вы используете последнюю версию tensorflow? это также можно решить при установке tf-nightly. Или компилируем тензорный поток из исходников с помощью bazel. Вы также можете использовать вместо этого: tf.contrib.lite.TocoConverter.from_frozen_graph(final_model_path, input_arrays, output_arrays)
Предоставьте график тензорного потока, который вы пытаетесь преобразовать, чтобы воспроизвести точную ошибку. Также было бы полезно сообщить о проблеме на GitHub https://github.com/tensorflow/tensorflow/issues < / а>
