File size: 653 Bytes
3135a01 d3d4eba 3135a01 d3d4eba 3135a01 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
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')
|