Android Tensflow GPU委托

时间:2020-03-18 07:33:50

标签: java android tensorflow

我是Android上tflite实现的新手。我正在使用此代码->

 delegate = new GpuDelegate();
 Interpreter.Options options = (new Interpreter.Options()).addDelegate(delegate);
 options.setNumThreads(6);
 tflite = new Interpreter(loadModelFile((Activity) context, modelFile), options);

要在android中使用GPU委托,请从tflite的官方文档中查找。代码非常好,并且比我仅使用CPU处理之前提供了更好的性能,但是我不知道setNumThreads()方法中可以传递的最大数量应该是多少,这样我的处理时间应该尽可能少。在这里,我只通过了6,但是不知道实际应该是什么。任何人都可以详细说明一下,setNumThreads()方法的含义是什么,以及如何以最喜欢的方式使用它来减少我的图像处理时间。

0 个答案:

没有答案