将数组从 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)))
.
.

我将不胜感激，

最好的。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）