Update main_test_SRMNet.py

main_test_SRMNet.py

@@ -22,8 +22,8 @@ tasks = ['Deblurring_motionblur',
 
 def main():
     parser = argparse.ArgumentParser(description='Quick demo Image Restoration')
-    parser.add_argument('--input_dir', default='test/', type=str, help='Input images root')
-    parser.add_argument('--result_dir', default='result/', type=str, help='Results images root')
+    parser.add_argument('--input_dir', default='./test/', type=str, help='Input images root')
+    parser.add_argument('--result_dir', default='./result/', type=str, help='Results images root')
     parser.add_argument('--weights_root', default='experiments/pretrained_models', type=str, help='Weights root')
     parser.add_argument('--task', default='Retouching', type=str, help='Restoration task (Above task list)')
 
@@ -63,7 +63,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('finish !')