Update main_test_SRMNet.py

main_test_SRMNet.py

@@ -26,8 +26,8 @@ def clean_folder(folder):
             
 def main():
     parser = argparse.ArgumentParser(description='Demo Image Denoising')
-    parser.add_argument('--input_dir', default='test/', type=str, help='Input images')
-    parser.add_argument('--result_dir', default='result/', type=str, help='Directory for results')
+    parser.add_argument('--input_dir', default='test', type=str, help='Input images')
+    parser.add_argument('--result_dir', default='result', type=str, help='Directory for results')
     parser.add_argument('--weights',
                         default='experiments/pretrained_models/AWGN_denoising_SRMNet.pth', type=str,
                         help='Path to weights')