import glob
import skimage
from skimage import io,transform, color
import os
import numpy as np
# *代表全部文件,*.png,*pdf,*jpg 可以选中特定类型
# 'C:\Users\mayuhua\Desktop\*\*.png' 桌面所有子文件夹内的png图像
input = r'C:\Users\mayuhua\Desktop\深度学习SR\图像数据集\DF2K\DIV2K_train_HR\*.png' # 选取文件,加上\*.tif
output = r'C:\Users\mayuhua\Desktop\深度学习SR\图像数据集\DF2K\DIV2K_train_灰度' # 输出路径
path_list = glob.glob(input) # 图像路径列表。
for path in path_list:
print(path) # 图像路径
base_name = os.path.basename(path) ; print(base_name) #文件名 name.tif
# base_name = os.path.splitext(base_name)[0] ; print(base_name) #文件名(无后缀) name
path_out = output+'\\'+base_name ; print(path_out) #输出路径 C:\Users\···\UCMerced_LandUse-图片集合\name.tif
img = io.imread(path) # 读取
img_out = color.rgb2gray(img) # 转为灰度
img_out = np.array(img_out, dtype=np.float32) # tif图片:需要由float64改为float32才能预览。png/jpg等格式,注释掉此行。
io.imsave(path_out, img_out) # 保存