Spaces:
Sleeping
Sleeping
File size: 1,572 Bytes
206ebf9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | from torchvision.transforms import Compose, ToTensor, RandomCrop, RandomHorizontalFlip, RandomVerticalFlip
from data.LOLdataset import *
from data.eval_sets import *
from data.SICE_blur_SID import *
from data.fivek import *
def transform1(size=256):
return Compose([
RandomCrop((size, size)),
RandomHorizontalFlip(),
RandomVerticalFlip(),
ToTensor(),
])
def transform2():
return Compose([ToTensor()])
def get_lol_training_set(data_dir,size):
return LOLDatasetFromFolder(data_dir, transform=transform1(size))
def get_lol_v2_training_set(data_dir,size):
return LOLv2DatasetFromFolder(data_dir, transform=transform1(size))
def get_training_set_blur(data_dir,size):
return LOLBlurDatasetFromFolder(data_dir, transform=transform1(size))
def get_lol_v2_syn_training_set(data_dir,size):
return LOLv2SynDatasetFromFolder(data_dir, transform=transform1(size))
def get_SID_training_set(data_dir,size):
return SIDDatasetFromFolder(data_dir, transform=transform1(size))
def get_SICE_training_set(data_dir,size):
return SICEDatasetFromFolder(data_dir, transform=transform1(size))
def get_SICE_eval_set(data_dir):
return SICEDatasetFromFolderEval(data_dir, transform=transform2())
def get_eval_set(data_dir):
return DatasetFromFolderEval(data_dir, transform=transform2())
def get_fivek_training_set(data_dir,size):
return FiveKDatasetFromFolder(data_dir, transform=transform1(size))
def get_fivek_eval_set(data_dir):
return SICEDatasetFromFolderEval(data_dir, transform=transform2()) |