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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。