Update test_sr.py

test_sr.py

@@ -320,14 +320,18 @@ def main(L_path, save_path, manual_label, use_real_ocr, use_new_bbox):
     parser.add_argument('-i', '--test_path', type=str, default='./Testsets/LQs')
     parser.add_argument('-o', '--save_path', type=str, default=None)
     parser.add_argument('-m', '--manual', action='store_true')
+    parser.add_argument('--real_ocr', action='store_true')
     args = parser.parse_args()
 
     '''
     We add new real-world ocr model and new robust bbox detection.
     Set 
     '''
     use_new_bbox = True
-    use_real_ocr = True
+    if args.real_ocr:
+        use_real_ocr = True
+    else:
+        use_real_ocr = False
 
     save_path = args.save_path
     if save_path is None:
@@ -347,8 +351,6 @@ def main(L_path, save_path, manual_label, use_real_ocr, use_new_bbox):
     else:
         print('{:>28s} : {}'.format('OCR Module', 'using ocr model trained on our synthetic data'))
 
-
-
 
     main(args.test_path, save_path, args.manual, use_real_ocr, use_new_bbox)