如何在分类模型中对单个图像进行预测？

Question

我遵循了一些COLAB，能够训练模型并评估其准确性。但是，它没有显示如何对单个输入进行预测。我只是想测试这个模型来预测一个新的例子，而不是一批例子

当我使用model.predict(image)其中图像是图像的NumPy数组时，我得到了这个错误

Error when checking input: expected keras_layer_input to have 4 dimensions, but got array with shape (2, 1)

如何预测单个图像并找到其类别

链接到my colab:https://colab.research.google.com/drive/1dBwo43ULo99FsUQoxYRsaATIpCANPtkh

Answer 1

该错误表示您尝试发送到模型的图像不是它期望的格式

您可能需要执行相同的预印（重塑， 缩放等）用于输入数据以及用于培训的数据

如果您分享培训代码，我们可以详细说明具体需要做什么

Answer 2

通过展开第一个维度，可以生成大小为1的批次：

image = np.expand_dims(image, 0)

也就是说，您的图像应该是(height, width, channels)或(channels, height, width)形状的，但从您收到的错误消息来看，情况并非如此