| from data.dataset import Dataset | |
| from utils import interact | |
| class Demo(Dataset): | |
| """Demo train, test subset class | |
| """ | |
| def __init__(self, args, mode='demo'): | |
| super(Demo, self).__init__(args, mode) | |
| def set_modes(self): | |
| self.modes = ('demo') | |
| def set_keys(self): | |
| super(Demo, self).set_keys() | |
| self.blur_key = '' # all the files | |
| self.non_sharp_keys = [''] # no files | |
| def __getitem__(self, idx): | |
| blur, sharp, pad_width, idx, relpath = super(Demo, self).__getitem__(idx) | |
| return blur, sharp, pad_width, idx, relpath | |