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:21:44 +08:00
parent 1150b8e846
commit 879d24e7fb
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
eval.py
View File

@ -16,7 +16,6 @@ def config_parser():
parser.add_argument('--snr', default=20, type=int, help='snr')
parser.add_argument('--test_image', default='./demo/kodim08.png', type=str, help='demo_image')
parser.add_argument('--times', default=100, type=int, help='num_workers')
parser.add_argument('--filters', default=40, type=int, help='channel type')
return parser.parse_args()
@ -38,7 +37,7 @@ def main():
psnr_all = 0.0
for i in range(args.times):
demo_image = model(test_image)
image = image_normalization('denormalization')(image)
demo_image = image_normalization('denormalization')(image)
gt = image_normalization('denormalization')(gt)
psnr_all += get_psnr(demo_image, test_image)
demo_image = torch.cat([test_image, demo_image], dim=1)