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