-
Notifications
You must be signed in to change notification settings - Fork 5
/
test_demoire.py
33 lines (25 loc) · 1001 Bytes
/
test_demoire.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import os
from os.path import join
import torch.backends.cudnn as cudnn
import data.sirs_dataset as datasets
from engine import Engine
from options.net_options.train_options import TrainOptions
from tools import mutils
opt = TrainOptions().parse()
opt.isTrain = False
cudnn.benchmark = True
opt.no_log = True
opt.display_id = 0
opt.verbose = False
datadir = os.path.join(os.path.expanduser('~'), 'datasets/demoire', 'eval')
eval_dataset = datasets.MoireDataset(datadir, phase='eval')
# eval_dataset = datasets.RealDataset(test_real_dir)
eval_dataloader = datasets.DataLoader(eval_dataset, batch_size=1, shuffle=False, num_workers=opt.nThreads,
pin_memory=True)
engine = Engine(opt)
# engine.save_eval(label='ytmt_ucs_demoire_opt_086_00860000')
# exit(0)
"""Main Loop"""
result_dir = os.path.join('./results', opt.name, mutils.get_formatted_time())
res = engine.eval(eval_dataloader, savedir=join(result_dir, 'demoire'), dataset_name='LCD2019')
print(res)