keras - 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

标签： kerasweb-deployment

Answer 1

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

尝试更改第 3 行

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

进入

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