52Hz commited on
Commit
8bdc2fe
1 Parent(s): c323f83

Update main_test_SRMNet.py

Browse files
Files changed (1) hide show
  1. main_test_SRMNet.py +1 -1
main_test_SRMNet.py CHANGED
@@ -19,7 +19,7 @@ def main():
19
  parser.add_argument('--input_dir', default='test/', type=str, help='Input images')
20
  parser.add_argument('--result_dir', default='result/', type=str, help='Directory for results')
21
  parser.add_argument('--weights',
22
- default='experiments/pretrained_models/AWGN_denoising_SRMNet.pth', type=str,
23
  help='Path to weights')
24
 
25
  args = parser.parse_args()
 
19
  parser.add_argument('--input_dir', default='test/', type=str, help='Input images')
20
  parser.add_argument('--result_dir', default='result/', type=str, help='Directory for results')
21
  parser.add_argument('--weights',
22
+ default='experiments/pretrained_models/real_denoising_SRMNet.pth', type=str,
23
  help='Path to weights')
24
 
25
  args = parser.parse_args()