')' is missed in line 32 of test_demo.py

')' is accidentally deleted in line 32 of test_demo.py
lidq92 · Aug 1, 2020 · f57d7b5 · f57d7b5
1 parent 3b77285
commit f57d7b5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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')