python图像将某通道颜色改变_在Python中，将3通道rgb彩色图像更改为1通道灰度有多快？...

我在一个4D数组中有将近40000个图像，其中包含原始像素数据(示例数、宽度、高度、通道数)。每幅图像的宽度为32像素，高度为32像素，RGB颜色有3个通道。我想将它们更改为灰度图像(从3个具有rgb的通道中选择1个具有强度)。我怎么能做得很快？

我的代码：import pickle

import cv2

training_file = "/train.p"

with open(training_file, mode='rb') as f:

train = pickle.load(f)

X_train = train['features']

def rgb2gray(rgb):

r, g, b = rgb[0], rgb[1], rgb[2]

gray = 0.2989 * r + 0.5870 * g + 0.1140 * b

return gray

X_train_gray = X_train.copy()

for i in range (X_train_gray.shape[0]):

for j in range (X_train_gray.shape[1]):

for k in range (X_train_gray.shape[2]):

rgb = X_train_gray[i,j,k]

gray = rgb2gray(rgb)

X_train_gray[i,j,k] = gray

print("X_train image data shape =", X_train.shape)

print("X_train