Spaces:
Running
Running
Update main_test_SRMNet.py
Browse files- main_test_SRMNet.py +1 -1
main_test_SRMNet.py
CHANGED
@@ -30,7 +30,7 @@ def main():
|
|
30 |
args = parser.parse_args()
|
31 |
|
32 |
# Prepare testing data
|
33 |
-
files = natsorted(glob(os.path.join(args.inp_dir, '*.PNG'))
|
34 |
if len(files) == 0:
|
35 |
raise Exception("\nNo images in {} \nPlease enter the following tasks: \n\n{}".format(inp_dir, '\n'.join(tasks)))
|
36 |
os.makedirs(args.result_dir, exist_ok=True)
|
|
|
30 |
args = parser.parse_args()
|
31 |
|
32 |
# Prepare testing data
|
33 |
+
files = natsorted(glob(os.path.join(args.inp_dir, '*.PNG'))
|
34 |
if len(files) == 0:
|
35 |
raise Exception("\nNo images in {} \nPlease enter the following tasks: \n\n{}".format(inp_dir, '\n'.join(tasks)))
|
36 |
os.makedirs(args.result_dir, exist_ok=True)
|