File size: 711 Bytes
3135a01 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from importlib import import_module
from torch.utils.data import dataloader
class Data:
def __init__(self, args):
self.loader_test = []
for d in args.data_test:
if d in ['Set5', 'Set14', 'B100', 'Urban100']:
m = import_module('data.benchmark')
testset = getattr(m, 'Benchmark')(args, name=d)
else:
raise NotImplementedError
self.loader_test.append(
dataloader.DataLoader(
testset,
batch_size=1,
shuffle=False,
pin_memory=False,
num_workers=args.n_threads,
)
)
|