52Hz commited on
Commit
2e96c8a
·
1 Parent(s): 9e97f39

Update main_test_SRMNet.py

Browse files
Files changed (1) hide show
  1. main_test_SRMNet.py +1 -2
main_test_SRMNet.py CHANGED
@@ -47,8 +47,7 @@ def main():
47
 
48
  os.makedirs(out_dir, exist_ok=True)
49
 
50
- files = natsorted(glob(os.path.join(inp_dir, '*.JPG'))
51
- + glob(os.path.join(inp_dir, '*.PNG')))
52
 
53
  if len(files) == 0:
54
  raise Exception(f"No files found at {inp_dir}")
 
47
 
48
  os.makedirs(out_dir, exist_ok=True)
49
 
50
+ files = natsorted(glob(os.path.join(inp_dir, '*.JPG')) + glob(os.path.join(inp_dir, '*.PNG')))
 
51
 
52
  if len(files) == 0:
53
  raise Exception(f"No files found at {inp_dir}")