def __init__(self, rgb_dir, target_transform=None):
super(DataLoaderTestSR, self).__init__()
self.target_transform = target_transform
LR_files = sorted(os.listdir(os.path.join(rgb_dir)))
self.LR_filenames = [os.path.join(rgb_dir, x) for x in LR_files if is_png_file(x)]
self.tar_size = len(self.LR_filenames)
def __init__(self, rgb_dir, target_transform=None):
super(DataLoaderVal, self).__init__()
self.target_transform = target_transform
clean_files = sorted(os.listdir(os.path.join(rgb_dir, 'groundtruth')))
noisy_files = sorted(os.listdir(os.path.join(rgb_dir, 'input')))
self.clean_filenames = [os.path.join(rgb_dir, 'groundtruth', x) for x in clean_files if is_png_file(x)]
self.noisy_filenames = [os.path.join(rgb_dir, 'input', x) for x in noisy_files if is_png_file(x)]
self.tar_size = len(self.clean_filenames)