| import os | |
| from data import srdata | |
| class Benchmark(srdata.SRData): | |
| def __init__(self, args, name='', benchmark=True, input_data_format ='NHWC'): | |
| super(Benchmark, self).__init__( | |
| args, name=name, benchmark=True, input_data_format=input_data_format | |
| ) | |
| def _set_filesystem(self, dir_data): | |
| self.apath = os.path.join(dir_data, 'benchmark', self.name) | |
| self.dir_hr = os.path.join(self.apath, 'HR') | |
| if self.input_large: | |
| self.dir_lr = os.path.join(self.apath, 'LR_bicubicL') | |
| else: | |
| self.dir_lr = os.path.join(self.apath, 'LR_bicubic') | |
| self.ext = ('', '.png') | |