Update main_test_SRMNet.py

main_test_SRMNet.py

@@ -31,7 +31,8 @@ def main():
 
     # Prepare testing data
     files = natsorted(glob(os.path.join(args.input_dir, '*.JPG')) + glob(os.path.join(args.input_dir, '*.PNG')))
-
+    if len(files) == 0:
+        raise Exception(f"No files found at {args.inp_dir}")
     os.makedirs(args.result_dir, exist_ok=True)
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     # Build model
@@ -63,7 +64,7 @@ def main():
 
         f = os.path.splitext(os.path.split(file_)[-1])[0]
         save_img((os.path.join(args.result_dir, f + '.png')), restored)
-        print(os.path.join(args.result_dir, f + '.png'))
+        print('{}'.format(os.path.join(args.result_dir, f + '.png')))
     print('finish !')