将灰度图转化为RGB图像
跑GAN的适合需要RGB图像,原图是灰度图,代码是网上找的,自己修改了一下
`# coding:UTF-8
from skimage import io, transform, color
import numpy as np
def convert_rgb(f, **args): # 图片处理与格式化的函数
gray = io.imread(f) # 读取图片
rgb = color.gray2rgb(gray) # 将灰度图片转换为彩色图片,实际上看上去还是灰度图
# dst=transform.resize(rgb,(64,64)) #调整大小,图像分辨率为64*64
return rgb
datapath = r'E:\jupyter notebook\OCT1\CNV' #待处理图像位置
str = datapath + '/*.jpeg' # 识别.jpeg的图像
coll = io.ImageCollection(str, load_func=convert_rgb) # 批处理
for i in range(len(coll)):
io.imsave(r'E:\jupyter notebook\OCT2\CNV' + np.str(i) + '.jpeg', coll[i]) # 保存图片在E:\jupyter notebook\OCT2\CNV