| | import os |
| |
|
| | |
| | from data import srdata |
| | import numpy as np |
| | import torch |
| | import torch.utils.data as data |
| |
|
| | class Benchmark(srdata.SRData): |
| | def __init__(self, args, name='', benchmark=True): |
| | super(Benchmark, self).__init__( |
| | args, name=name, benchmark=True |
| | ) |
| |
|
| | 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') |
| |
|
| |
|
