Tensorflow 字节缓冲区

时间:2021-05-13 10:48:34

标签: java android android-studio tensorflow machine-learning

我正在使用 TensorFlow 和 android studio 生成一个模型,当用户提供 1 到 4 的输入时,该模型将返回一个范围从 60 到 150 的数字。

当我将模型插入到我的 android studio 项目时生成的这段代码让我有点迷茫,因为我在 byteByffer 变量上有错误。 任何人都可以向我解释这是什么?我试图找到答案,但没有得到任何东西......我怀疑这是我需要提供给模型的输入,但是我需要以某种方式将其转换为 byteByffer ?非常感谢您的帮助!

try {
        BPMmodel model = BPMmodel.newInstance(this);

        // Creates inputs for reference.
        TensorBuffer inputFeature0 = TensorBuffer.createFixedSize(new int[]{1, 1}, DataType.FLOAT32);
        inputFeature0.loadBuffer(byteBuffer);

        // Runs model inference and gets result.
        BPMmodel.Outputs outputs = model.process(inputFeature0);
        TensorBuffer outputFeature0 = outputs.getOutputFeature0AsTensorBuffer();
        BPMfromModel = outputFeature0.getFloatValue();

        // Releases model resources if no longer used.
        model.close();
    } catch (IOException e) {
        // TODO Handle the exception
    }

0 个答案:

没有答案