python实现灰度图转伪BGR图

可批量处理单通道的灰度图转成BGR图,用于深度学习训练

python代码:

from PIL import Image
import numpy as np
import os
import cv2

img_dir='./concrete_4_predict_data_gray/v_215_g01_c01/'#图片所在位置    
save_dir='./concrete_4_predict_data_gray/v_215_g01_c01/'#保存图片的位置

for img_name in os.listdir(img_dir):
    img_path = img_dir + img_name
    print('正在处理',img_name)
    img = Image.open(img_path)
    fake_rgb = img.convert('RGB')
    fake_bgr=cv2.cvtColor(np.asarray(fake_rgb),cv2.COLOR_RGB2BGR)
    cv2.imwrite(save_dir + img_name[0:-4] + '.jpg', fake_bgr)
    cv2.COLOR_GRAY2BGR

你可能感兴趣的:(python)