import argparseimport globimport h5pyimport numpy as npimport PIL.Image as pil_imagefrom utils import calc_patch_size, convert_rgb_to_yfrom PIL import Imageimport randomimport osdef random_crop(image, crop_shape):###image 是pil读取的，crop_shape是裁剪的大小nw = random.randint(0, image.size[0] - crop_shape[0]) ##裁剪图像在原图像中的坐标nh = random.randint(0, image.size[1] - crop_shape[1])image_crop = image.crop((nh, nw, nh + crop_shape[0], nw + crop_shape[1]))return image_cropdef convert_rgb_to_y(img, dim_order=hwc):if dim_order == hwc:return 16. + (64.738 * img[..., 0] + 129.057 * img[..., 1] + 25.064 * img[..., 2]) / 256.else:return 16. + (64.738 * img[0] + 129.057 * img[1] + 25.064 * img[2]) / 256.def train(args):h5_file = h5py.File(args.output_path, w)lr_patches = []hr_patches = []
