def get_image_set(self, image_hr_list): y_imgs = [] x_imgs = [] # use 10 threads to read files imgs_hr = tl.visualize.read_images(image_hr_list, self.groundtruth_dir) imgs_lr = self.__downsample_imgs(imgs_hr, interp=self.interp, scale=self.scale) for i in range(len(imgs_lr)): #crop the image randomly try: x_img,y_img = utils.crop(imgs_lr[i], imgs_hr[i], self.image_size, self.image_size, self.scale, is_random=True) except LargeSizeException as e: print(e) else: y_imgs.append(y_img) x_imgs.append(x_img) return x_imgs, y_imgs
def get_image_set(self, image_lr_list,input_dir,ground_truth_dir, postfix_len): y_imgs = [] x_imgs = [] # use 10 threads to read files imgs_lr = tl.visualize.read_images(image_lr_list, input_dir) image_hr_list = utils.get_hrimg_list(image_lr_list, postfix_len) imgs_hr = tl.visualize.read_images(image_hr_list, ground_truth_dir) for i in range(len(imgs_lr)): #crop the image randomly try: x_img,y_img = utils.crop(imgs_lr[i], imgs_hr[i], self.image_size, self.image_size, self.scale, is_random=True) except LargeSizeException as e: print(e) else: y_imgs.append(y_img) x_imgs.append(x_img) return x_imgs, y_imgs