Python tf.lite.experimental.load_delegate用法及代码示例

返回加载的委托对象。

用法

tf.lite.experimental.load_delegate(
    library, options=None
)

参数

library 包含的共享库的名称TfLiteDelegate.
options 加载委托所需的选项字典。字典中的所有键和值都应该可以转换为 str。查阅特定代表的文档以获取所需的合法选项。 (默认无)

委托对象。

抛出

ValueError 代理加载失败。
RuntimeError 如果在不受支持的平台上使用委托加载。

示例用法：

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/

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.lite.experimental.load_delegate。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。