Android上的Tensorflow错误

Question

我正在学习 Tensorflow，并按照教程制作了一个自定义模型以在 Android 应用程序中运行它，但我遇到了问题。我有以下代码：

    public void testModel(Context ctx) {
        String model_file = "file:///android_asset/model_graph.pb";
        int[] result = new int[2];
        float[] input = new float[]{0.0F, 1.0F, 0.0F, 1.0F, 1.0F, 0.0F, 0.0F, 1.0F, 0.0F, 1.0F, 0.0F, 1.0F, 0.0F, 1.0F, 1.0F, 0.0F, 0.0F, 0.0F, 0.0F, 1.0F, 1.0F, 0.0F, 1.0F, 0.0F, 1.0F, 0.0F, 1.0F, 0.0F, 0.0F, 1.0F, 1.0F, 0.0F, 0.0F, 0.0F, 1.0F, 0.0F, 0.0F, 1.0F, 0.0F, 1.0F, 0.0F, 1.0F, 1.0F, 0.0F, 0.0F, 1.0F, 0.0F, 0.0F, 0.0F, 1.0F, 0.0F, 1.0F, 0.0F, 1.0F, 1.0F, 0.0F, 1.0F, 0.0F, 0.0F, 0.0F, 1.0F, 0.0F, 1.0F, 0.0F, 1.0F, 0.0F, 1.0F, 0.0F};
        TensorFlowInferenceInterface inferenceInterface;
        inferenceInterface = new TensorFlowInferenceInterface(ctx.getAssets(), model_file);
        inferenceInterface.feed("input", input, 68);
        inferenceInterface.run(new String[]{"output"});
        inferenceInterface.fetch("output", result);
        Log.v(TAG, Arrays.toString(result));
    }

当应用尝试运行inferenceInterface.run(new String[]{"output"}) 方法时出现错误：

java.lang.IllegalArgumentException: In[0] is not a matrix
[[Node: MatMul = MatMul[T=DT_FLOAT, transpose_a=false, transpose_b=false, _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_input_0_0, W1)]]

我不相信我创建的模型是问题所在，因为我能够在 Python 代码中使用它并获得肯定的结果。

Answer 1

从错误消息 (In[0] is not a matrix) 看来，您的模型要求输入是一个矩阵（即二维张量），而您正在输入一个包含 68 个元素的一维张量（向量）。

特别是，TensorFlowInferenceInterface.feed 的 dims 参数在该行中似乎不正确：

inferenceInterface.feed("input", input, 68);

相反，它应该是这样的：

inferenceInterface.feed("input", input, 68, 1);

如果您的模型需要 68x1 矩阵（或 34, 2 如果需要 34x2 矩阵，17, 4 需要 17x4 矩阵等）

希望对您有所帮助。