如何将RGB图像数据集转换为单通道灰度？

Question

我遇到的问题是，当我使用灰度图像数据集输入我的cnn时，它将其视为3通道图像而不是单通道。

那么如何将3通道图像转换为单通道灰度图像呢？

Answer 1

您可以使用以下方式直接将图像读取为灰度：

im_gray = cv2.imread('gray_image.png', cv2.IMREAD_GRAYSCALE)

或者您可以使用以下命令将rgb图像转换为灰度：

img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)