Update main_test_CMFNet.py

main_test_CMFNet.py

@@ -14,11 +14,11 @@ from natsort import natsorted
 from model.CMFNet import CMFNet
 
 def main():
-    parser = argparse.ArgumentParser(description='Demo Image Deraindrop')
+    parser = argparse.ArgumentParser(description='Demo Image Dehaze')
     parser.add_argument('--input_dir', default='test/', type=str, help='Input images')
     parser.add_argument('--result_dir', default='results/', type=str, help='Directory for results')
     parser.add_argument('--weights',
-                        default='experiments/pretrained_models/deraindrop_model.pth', type=str,
+                        default='experiments/pretrained_models/dehaze_model.pth', type=str,
                         help='Path to weights')
 
     args = parser.parse_args()