python - Tensorflow Keras 预测返回错误形状的输出

Question

我正在尝试预测单个图像。但是我的模型在应该是 (1,10) 时返回一个形状为 (1,1,1,2048) 的预测数组。知道我做错了什么吗？我的 x 输入形状在 (1,32,32,3) 处是正确的。

ResNet50V2(): 
  IMG_SHAPE = (32, 32, 3)
  return tf.keras.applications.ResNet50V2(input_shape=IMG_SHAPE, include_top=False, weights=None, classes=10)

model = ResNet50V2()
x = x[None, :]
predictions = model.predict(x)

标签： pythontensorflowkerasconv-neural-networkresnet

Answer 1

您正在使用参数加载您的 keras 模型

include_top=False

负责将模型输出投影到您预期的类数量的全连接投影层的切割。将参数更改为 True。