hc/JSCC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

eval.py modified

Browse Source
This commit is contained in:
chun 2023-12-23 22:22:16 +08:00
parent 879d24e7fb
commit c7baea5a24
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
eval.py
View File

@ -37,9 +37,9 @@ def main():
psnr_all = 0.0
for i in range(args.times):
demo_image = model(test_image)
demo_image = image_normalization('denormalization')(image)
gt = image_normalization('denormalization')(gt)
psnr_all += get_psnr(demo_image, test_image)
demo_image = image_normalization('denormalization')(demo_image)
gt = image_normalization('denormalization')(test_image)
psnr_all += get_psnr(demo_image, gt)
demo_image = torch.cat([test_image, demo_image], dim=1)
demo_image = transforms.ToPILImage()(demo_image)
demo_image.save('./demo/demo.png')