| import os |
| from data import srdata |
| from data import div2k |
|
|
| class DIV2KJPEG(div2k.DIV2K): |
| def __init__(self, args, name='', train=True, benchmark=False): |
| self.q_factor = int(name.replace('DIV2K-Q', '')) |
| super(DIV2KJPEG, self).__init__( |
| args, name=name, train=train, benchmark=benchmark |
| ) |
|
|
| def _set_filesystem(self, dir_data): |
| self.apath = os.path.join(dir_data, 'DIV2K') |
| self.dir_hr = os.path.join(self.apath, 'DIV2K_train_HR') |
| self.dir_lr = os.path.join( |
| self.apath, 'DIV2K_Q{}'.format(self.q_factor) |
| ) |
| if self.input_large: self.dir_lr += 'L' |
| |
| self.ext = ('.tif', '.jpg') |
|
|
|
|
