我在使用CNN模型进行预测时遇到了问题

模型结构

from tensorflow.keras.preprocessing import imageimport numpy as npimg = image.load_img("test/apple/apple.jpg", target_size=(150,150))x=image.img_to_array(img) / 255x = x.reshape(1,-1)model.predict(x)

回答：

您将图像展平了，但您的模型需要批量处理图像数据。添加一个维度，使用np.expand_dims到调整大小的图像上，然后传递给模型进行预测。

尝试这样做。

img = image.load_img("test/apple/apple.jpg", target_size=(150,150))x=image.img_to_array(img) / 255resized_img_np = np.expand_dims(x,axis=0)prediction = model.predict(resized_img_np)