导入
使用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)