52Hz commited on
Commit
8938041
·
1 Parent(s): 4619bfc

Update main_test_CMFNet.py

Browse files
Files changed (1) hide show
  1. main_test_CMFNet.py +1 -1
main_test_CMFNet.py CHANGED
@@ -16,7 +16,7 @@ from model.CMFNet import CMFNet
16
  def main():
17
  parser = argparse.ArgumentParser(description='Demo Image Deraindrop')
18
  parser.add_argument('--input_dir', default='test/', type=str, help='Input images')
19
- parser.add_argument('--result_dir', default='result/', type=str, help='Directory for results')
20
  parser.add_argument('--weights',
21
  default='experiments/pretrained_models/deraindrop_model.pth', type=str,
22
  help='Path to weights')
 
16
  def main():
17
  parser = argparse.ArgumentParser(description='Demo Image Deraindrop')
18
  parser.add_argument('--input_dir', default='test/', type=str, help='Input images')
19
+ parser.add_argument('--result_dir', default='results/', type=str, help='Directory for results')
20
  parser.add_argument('--weights',
21
  default='experiments/pretrained_models/deraindrop_model.pth', type=str,
22
  help='Path to weights')