将RGB图像转换为灰度图像

Question

我有一个numpy的rgb图像数组，如下所示

print(train_frames.shape) #(100, 40, 120, 140, 3)

这意味着40帧中的每100行包含120x140的图像。

我需要将其转换为灰度图像，即

print(train_frames.shape) #(100, 40, 120, 140, 1)

如何转换？

Answer 1

如果要批量执行此操作，一种方法是使用numpy.mean

numpy.mean(train_frames, axis=-1)

axis = -1表示3个rgb频道。

如果您想获得更好的转换，我想您需要使用opencv cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

逐个图像地进行转换。