CUB-200-2011 数据集灰度图片列表、转三通道 python 代码

import cv2
from PIL import Image
import numpy as np

the_dir = './CUB/'
imgs = [the_dir + '063.Ivory_Gull/Ivory_Gull_0040_49180.jpg',
        the_dir + '093.Clark_Nutcracker/Clark_Nutcracker_0020_85099.jpg',
        the_dir + '025.Pelagic_Cormorant/Pelagic_Cormorant_0022_23802.jpg',
        the_dir + '066.Western_Gull/Western_Gull_0002_54825.jpg',
        the_dir + '009.Brewer_Blackbird/Brewer_Blackbird_0028_2682.jpg',
        the_dir + '063.Ivory_Gull/Ivory_Gull_0085_49456.jpg',
        the_dir + '087.Mallard/Mallard_0130_76836.jpg',
        the_dir + '108.White_necked_Raven/White_Necked_Raven_0070_102645.jpg',]

for img in imgs:
    I = Image.open(img)
    L = I.convert('L')
    a = np.array(L)
    image = np.expand_dims(a, axis=2)
    image = np.concatenate((image, image, image), axis=-1)
    cv2.imwrite(img, image)