问在Android中运行tflite分类器(Kotlin)

EN

import tensorflow as tf
import numpy as np
interpreter = tf.lite.Interpreter(model_path="model.tflite")
interpreter.allocate_tensors()

# Get input and output tensors.
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()

# Test the model on random input data.
input_data = np.array([[566, 12, 12, -12, 1, 7, 1, 1, 1, -1]], dtype=np.float32)
interpreter.set_tensor(input_details[0]['index'], input_data)

interpreter.invoke()

# The function `get_tensor()` returns a copy of the tensor data.
# Use `tensor()` in order to get a pointer to the tensor.
[[output_data]] = interpreter.get_tensor(output_details[0]['index'])
print(output_data) # prints 0.99999845

val model = Model.newInstance(this)
val inputFeature0 = TensorBuffer.createFixedSize(intArrayOf(1, 10), DataType.FLOAT32)
var byteBuffer = ByteBuffer.allocate(4 * 10)
for (value in floatArrayOf(566f, 12f, 12f, -12f, 1f, 7f, 1f, 1f, 1f, -1f)) {
   byteBuffer.putFloat(value)
}
inputFeature0.loadBuffer(byteBuffer)
val outputs = model.process(inputFeature0)
val result = outputs.outputFeature0AsTensorBuffer.floatArray[0]
print(result) // 0.51473397
model.close()