Update main_test_SRMNet.py

main_test_SRMNet.py

@@ -34,7 +34,7 @@ def main():
     files = natsorted(glob.glob(os.path.join(inp_dir, '*')))
     if len(files) == 0:
         raise Exception("\nNo images in {} \nPlease enter the following tasks: \n\n{}".format(inp_dir, '\n'.join(tasks)))
-    os.makedirs(out_dir, exist_ok=True)
+    os.makedirs(args.result_dir, exist_ok=True)
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     # Build model
     model = define_model(args)