使用inferenceInterface.Feed()时如何处理多个输入?

时间:2019-04-07 15:54:08

标签: java android tensorflow

我有一个训练有素的张量流模型,现在我需要将其移植到Android平台上。但是我使用inferenceInterface.feed()函数有问题。在我发现的示例代码中,该函数只有一个输入节点。

    private static final String INPUT_NODE = "input:0"; // input tensor name
    private static final String OUTPUT_NODE = "output:0"; // output tensor name
    private static final String[] OUTPUT_NODES = {"output:0"}; 
    private static final int OUTPUT_SIZE = 10; // number of classes
    private static final int INPUT_SIZE = 784; // size of the input
    INPUT_IMAGE //MNIST Image
    float[] result = new float[OUTPUT_SIZE]; // get the output probabilities for each class

    inferenceInterface.feed(INPUT_NODE, INPUT_IMAGE, 1, INPUT_SIZE); //1-D input (1,INPUT_SIZE)
    inferenceInterface.run(OUTPUT_NODES);
    inferenceInterface.fetch(OUTPUT_NODE, result);

当我需要使用多个输入时,应该如何使用inferenceInterface.feed函数?我应该为每个输入调用一次此功能吗?如果尺寸不同,该怎么办?

0 个答案:

没有答案