将数组从 RGB 转换为 GRAYSCALE 图像

如何解决将数组从 RGB 转换为 GRAYSCALE 图像

当我运行下面的代码时，我遇到了重塑问题，这让我想知道问题出在哪里。它说; “无法将数组大小 100352000 重塑为形状 (224,224,3)”。可能是因为对RGB图像的预测，但是模型实际上需要灰度图像。

IMG_SIZE = 224
.
.
def process_data(data_image_list,DATA_FOLDER,isTrain=True):
    data_df = []
    for img in tqdm(data_image_list):
        path = os.path.join(DATA_FOLDER,img)
        if (isTrain):
            label = label_pet_image_one_hot_encoder(img)
        else:
            label = img.split('.')[0]
        img = cv2.imread(path,cv2.IMREAD_GRAYSCALE)
        img = cv2.resize(img,(IMG_SIZE,IMG_SIZE))
        data_df.append([np.array(img),np.array(label)])
    shuffle(data_df)
    return data_df
.
.
X = np.array([i[0] for i in train]).reshape(-1,IMG_SIZE,3)
y = np.array([i[1] for i in train])
.
.
model = Sequential()
model.add(Conv2D(16,(3,3),activation='relu',input_shape=(224,3)))
model.add(MaxPooling2D((2,2)))
.
.

我将不胜感激，

最好的。