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