File size: 688 Bytes
8ec10cf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | 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
|