是否可以为RGB和灰度图像创建训练矩阵

我正在为CNN创建培训矩阵。 图像既有RGB又有灰度。

创造出 [ # of images, #features ]

图像大小为:

1024* 1024

以下是我的代码:

from skimage.transform import rescale, resize
from skimage import io

features = np.empty((0,1024 * 1024), np.float32)
imagePath = directoyPath+"/"+ imageName
image = io.imread(imagePath)
print(image.shape)
flatFeatures = np.reshape(image,(1,1024*1024))
print(flatFeatures.shape)
features = np.append(features, flatFeatures, axis=0)
print(features.shape)

问题是RGB的形状是(10241024,3)。

如何将RGB和灰度图像传送到 features 矩阵。