android - 如何测试具有多个输入的 Tensorflowlite 模型？

Question

我创建了一个具有 4 个输入和一个输出的简单 MLP 回归 Keras 模型。我现在将这个模型转换为 TFlite，我只是想了解如何在 android studio 上测试它。如何输入多个 4D 对象以在 Java 中进行测试？尝试运行模型时出现以下错误:

try{
            tflite = new Interpreter(loadModelFile());
        }
        catch(Exception ex){
            ex.printStackTrace();
        }   

double[][] inp= new double[1][4];
    inp[0][1]= 0;
    inp[0][0] = 0;
    inp[0][2]= 0;
    inp[0][3]=-2.01616982303105;

    double[] output = new double[100];

    tflite.run(inp,output);

编辑:这是我最初创建的模型:

# create model
model = Sequential()
model.add(Dense(50, activation="tanh", input_dim=4, 
kernel_initializer="random_uniform", name="input_tensor"))
model.add(Dense(50, activation="tanh", 
kernel_initializer="random_uniform"))
model.add(Dense(1, activation="linear", 
kernel_initializer='random_uniform', name="output_tensor"))

Answer 1

如果您的输入实际上是 4 个独立的张量，那么您应该使用允许多个独立输入的 Interpreter.runForMultipleInputsAndOutputs API。另见 this example来自 TensorFlow Lite 存储库。例如:

double[] input0 = {...};
double[] input1 = {...};
Object[] inputs = {input0, input1};
double[] output = new double[100];
Map<Integer, Object> outputs = new HashMap<>();
outputs.put(0, output);
interpreter.runForMultipleInputsOutputs(inputs, outputs);