|
|
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') |
|
|
|
|
|
|
