Hugging Face's logo

Spaces:
52Hz
/
SRMNet_real_world_denoising
Runtime error

App Files Community
52Hz commited on
Commit
1eb1887
·
1 Parent(s): 2e96c8a

Update main_test_SRMNet.py

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