Spaces:
Runtime error
Runtime error
Update main_test_SRMNet.py
Browse files- 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/
|
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()
|