import glob
import numpy as np import cv2 import os from PIL import Image from scipy import misc from scipy import ndimage
“”“该程序作用是将单通道图像转成3通道”"" def expand(i_dir,o_dir): for name in os.listdir(i_dir): print(name) image_path = os.path.join(i_dir,name) img = cv2.imread(image_path,0) image = np.expand_dims(img, axis=2) image = np.concatenate((image, image, image), axis=-1) base_name = name.split(’.’)[0] #print(base_name) cv2.imwrite(o_dir+base_name+’.png’,image)
