ValueError：`decode_predictions`需要一批预测（即2D形状的数组（样本，1000个））。找到形状为（1，3）的数组

Question

enter image description here

我去过Google很久了，但是他们问的问题与我不同。他们使用现成的模型。我是一个自我训练的模型，但同样的错误让我感到困惑。

Answer 1

问题是decode_predictions期望类数为1000（即ImageNet数据集中的类数）。看来您的自定义模型只有3个输出类：

  

decode_predictions需要一批预测（即2D数组   形状(samples, 1000)）。找到形状为(1, 3)的数组。