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()
|