LeNet 神经网络模型报错 'shape '[1, 28, 28]' is invalid for input of size 3136' 解决方法

这个错误提示表明输入数据的形状不匹配，期望的形状是 '[1, 28, 28]'，但实际输入的形状大小为 3136。这可能是因为输入数据的形状没有被正确处理。

要解决这个问题，可以检查输入数据的形状是否正确，并确保在将数据输入到模型之前将其转换为正确的形状。如果输入数据的形状不正确，可以使用 reshape 函数将其转换为正确的形状。

例如，在使用 LeNet 模型处理 MNIST 手写数字数据集时，输入数据应该是 28x28 像素的灰度图像。因此，如果使用 OpenCV 读取图像数据，则需要将其转换为 28x28 的灰度图像，并将其转换为 numpy 数组，然后将其形状从 [28, 28] 转换为 [1, 28, 28]。可以使用以下代码进行转换：

import cv2
import numpy as np

# 读取图像数据
img = cv2.imread('example.png', cv2.IMREAD_GRAYSCALE)

# 重新调整图像大小
img = cv2.resize(img, (28, 28))

# 转换为 numpy 数组
img = np.array(img)

# 将形状从 [28, 28] 转换为 [1, 28, 28]
img = img.reshape(1, 28, 28)

# 将输入数据输入到 LeNet 模型中进行预测
result = model.predict(img)

这样，就可以将输入数据的形状正确地转换为 LeNet 模型所期望的形状，并避免出现“shape '[1, 28, 28]' is invalid for input of size 3136”这个错误。