tf.lite.experimental.load_delegate
Возвращает загруженный объект Delegate.
tf.lite.experimental.load_delegate(
library, options=None
)
Пример использования:
import tensorflow as tf
try:
delegate = tf.lite.experimental.load_delegate('delegate.so')
except ValueError:
// Fallback to CPU
if delegate:
interpreter = tf.lite.Interpreter(
model_path='model.tflite',
experimental_delegates=[delegate])
else:
interpreter = tf.lite.Interpreter(model_path='model.tflite')
Это обычно используется для использования EdgeTPU для запуска моделей TensorFlow Lite. Для получения дополнительной информации см.: https://coral.ai/docs/edgetpu/tflite-python/
| Аргументы | |
|---|---|
library | Имя общей библиотеки, содержащей TfLiteDelegate. |
options | Словарь параметров, необходимых для загрузки делегата. Все ключи и значения в словаре должны быть преобразуемы в строку. Обратитесь к документации конкретного делегата для получения необходимых и допустимых параметров. (по умолчанию None) |
| Возвращаемое значение | |
|---|---|
| Объект Delegate. |
| Исключения | |
|---|---|
ValueError | Делегат не удалось загрузить. |
RuntimeError | Если загрузка делегата используется на платформе, которая не поддерживается. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/lite/experimental/load_delegate