減少python中的圖像通道

Question

我有一個尺寸為 (128, 19, 3) 的圖像，我想將其轉換為 (128, 19, 1)。 我使用了此代碼（在 python 中），但它將圖像大小轉換為 (128, 19) 而不是我想要的 (128, 19, 1)。 謝謝如果有人可以幫忙

from PIL import Image
import glob
images = glob.glob('D:\\thesis\\Paper 3\\Feature 
Extraction\\two_dimension_Feature_extraction\\stft_feature\\Training_set\\P300\\*.png')
img = Image.open(images[0]).convert('L')

Answer 1

只需在數組末尾添加一個維度：

img = img[...,None]