| from data.dataset import Dataset | |
| from utils import interact | |
| class GOPRO_Large(Dataset): | |
| """GOPRO_Large train, test subset class | |
| """ | |
| def __init__(self, args, mode='train'): | |
| super(GOPRO_Large, self).__init__(args, mode) | |
| def set_modes(self): | |
| self.modes = ('train', 'test') | |
| def set_keys(self): | |
| super(GOPRO_Large, self).set_keys() | |
| self.blur_key = 'blur_gamma' | |
| # self.sharp_key = 'sharp' | |
| def __getitem__(self, idx): | |
| blur, sharp, pad_width, idx, relpath = super(GOPRO_Large, self).__getitem__(idx) | |
| relpath = relpath.replace('{}/'.format(self.blur_key), '') | |
| return blur, sharp, pad_width, idx, relpath | |