From f57d7b5fa7d9f6f2a177af28f1b486dd5773cf91 Mon Sep 17 00:00:00 2001 From: lidq92 Date: Sat, 1 Aug 2020 10:25:32 +0800 Subject: [PATCH] ')' is missed in line 32 of test_demo.py ')' is accidentally deleted in line 32 of test_demo.py --- README.md | 2 ++ test_demo.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 87c44c7..863277d 100644 --- a/README.md +++ b/README.md @@ -55,6 +55,8 @@ CUDA_VISIBLE_DEVICES=1 python test_dataset.py --dataset CLIVE --resize --arch re ```bash CUDA_VISIBLE_DEVICES=0 python test_demo.py --img data/1000.JPG --resize --arch resnext101_32x8d --trained_model_file checkpoints/p1q2.pth # > The image quality score is 10.430044178601875 +CUDA_VISIBLE_DEVICES=1 python test_demo.py --img data/1000.JPG --resize --arch resnext101_32x8d --trained_model_file checkpoints/p1q2plus0.1variant.pth +# > The image quality score is 16.726127839961094 ``` ### Remark diff --git a/test_demo.py b/test_demo.py index 56f09ef..8dc520c 100644 --- a/test_demo.py +++ b/test_demo.py @@ -29,7 +29,7 @@ def run(args): y = model(im.unsqueeze(0)) k = checkpoint['k'] b = checkpoint['b'] - print('The image quality score is {}'.format(y[-1].item() * k[-1] + b[-1]) + print('The image quality score is {}'.format(y[-1].item() * k[-1] + b[-1])) if __name__ == "__main__": parser = ArgumentParser(description='Test Demo for LinearityIQA')