holmesshuan / edsr-ssim Goto Github PK

Thank you for your perfect code.
I add your code to EDSR project ,but I see I can not calculate psnr and ssim at the same time,because ckp is just one.
I try to deepcopy it ,but it can not be copied.
Can you help me?

def test(self):
torch.set_grad_enabled(False)
epoch = self.optimizer.get_last_epoch()
self.ckp.write_log('\nEvaluation:')
self.ckp.add_log(
torch.zeros(1, len(self.loader_test), len(self.scale))
)
self.model.eval()
timer_test = utility.timer()
if self.args.save_results: self.ckp.begin_background()
#results_path = './results/' + cfg['sub_name'] + '/ICCV_now'
for idx_data, d in enumerate(self.loader_test):
for idx_scale, scale in enumerate(self.scale):
d.dataset.set_scale(idx_scale)
for lr, hr, filename in tqdm(d, ncols=80):
lr, hr = self.prepare(lr, hr)
sr = self.model(lr, idx_scale)
sr = utility.quantize(sr, self.args.rgb_range)
save_list = [sr]
self.ckp.log[-1, idx_data, idx_scale] += utility.calc_ssim(sr, hr, scale, self.args.rgb_range, dataset=d)
print(self.ckp.log)
if self.args.save_gt:
save_list.extend([lr, hr])
if self.args.save_results:
self.ckp.save_results(d, filename[0], save_list, scale)
self.ckp.log[-1, idx_data, idx_scale] /= len(d)
best = self.ckp.log.max(0)
print("best",best)
self.ckp.write_log(
'[{} x{}]\tSSIM: {:.3f} (Best: {:.3f} @epoch {})'.format(
d.dataset.name,
scale,
self.ckp.log[-1, idx_data, idx_scale],
best[0][idx_data, idx_scale],
best[1][idx_data, idx_scale] + 1
)
)

Hi. Thank you for your guidance and complete explanations.
I have done these changes in the code and it was run without error, but the problem I have is that the value of SSIM was not printed in the output, and I want to know what I should do if I want this value to be printed in the output.
thanks.