|
from data.dataset import Dataset
|
|
|
|
from utils import interact
|
|
|
|
class REDS(Dataset):
|
|
"""REDS train, val, test subset class
|
|
"""
|
|
def __init__(self, args, mode='train'):
|
|
super(REDS, self).__init__(args, mode)
|
|
|
|
def set_modes(self):
|
|
self.modes = ('train', 'val', 'test')
|
|
|
|
def set_keys(self):
|
|
super(REDS, self).set_keys()
|
|
|
|
|
|
|
|
self.non_blur_keys = ['blur', 'blur_comp', 'blur_bicubic']
|
|
self.non_blur_keys.remove(self.blur_key)
|
|
self.non_sharp_keys = ['sharp_bicubic', 'sharp']
|
|
self.non_sharp_keys.remove(self.sharp_key)
|
|
|
|
def __getitem__(self, idx):
|
|
blur, sharp, pad_width, idx, relpath = super(REDS, self).__getitem__(idx)
|
|
relpath = relpath.replace('{}/{}/'.format(self.mode, self.blur_key), '')
|
|
|
|
return blur, sharp, pad_width, idx, relpath
|
|
|
