Logo
Section 1 / 7

使用TF Lite模型进行推理

学习在 Android 应用中使用 TF Lite 解释器进行推理。

导入

使用TF Lite模型进行推理

学习在 Android 应用中使用 TF Lite 解释器进行推理。

...
使用 TF Lite 解释器执行推理的步骤

代码示例

1import org.tensorflow.lite.Interpreter
1val interpreter: Interpreter by lazy { 2 val options = Interpreter.Options() 3 options.setNumThreads(4) // 设置用于推理的线程数量 4 options.setUseNNAPI(true) // 启用 NNAPI 加速;若不可用则会报错 5 Interpreter(loadModelFile(), options) 6}

代码示例

1interpreter.run(input, output)

TensorFlow Lite

应用结构