python - 从 PIL 图像转换后，Numpy 数组不显示灰度图像的颜色尺寸

Question

我正在尝试使用以下代码片段将 RGB 图像转换为灰度图像，然后转换为 numpy 数组:

img = Image.open("image1.png")
img = img.convert('L')
img = np.array(img, dtype='f')
print(img.shape)

结果是形状为 (128, 128) 的 numpy 数组。无论如何，我是否可以将灰度图像转换为 numpy 数组，以便它也具有颜色 channel ，即形状为 (128, 128, 1)？

Answer 1

就像@Mark在评论中提到的那样，如果您的数组使用newaxis，则在末尾添加一个维度:

img=img[...,None]

None 的作用与 np.newaxis 类似。它不会创建颜色，而是添加类似于单 channel 图像的尺寸。