ValueError：无法将大小为50176的数组重塑为形状（224,224,3）

Question

我有一个错误“无法将大小为50176的数组重塑为形状（224,224,3），我看不出为什么会发生这种情况，也许有人遇到了同样的问题，

def model_predict(img_path,model):
    size=224
    img = cv2.imread(img_path,cv2.IMREAD_GRAYSCALE)

    img = cv2.resize(img,(size,size))

    img=np.array(img).reshape(-1,size,size,3)
    img_class = model.predict_classes(img)
    return img_class

Answer 1

您的图像是灰度（单通道），但是您尝试将其重塑为RGB（3通道），这是行不通的。

尝试更改第3行

img = cv2.imread(img_path,cv2.IMREAD_GRAYSCALE)

进入

img = cv2.imread(img_path,cv2.IMREAD_COLOR)