52Hz commited on
Commit
a4da779
·
1 Parent(s): f494bd6

Update main_test_SRMNet.py

Browse files
Files changed (1) hide show
  1. main_test_SRMNet.py +1 -1
main_test_SRMNet.py CHANGED
@@ -64,7 +64,7 @@ def main():
64
  restored = img_as_ubyte(restored[0])
65
 
66
  f = os.path.splitext(os.path.split(file_)[-1])[0]
67
- save_img((os.path.join(out_dir, f + '.png')), restored)
68
 
69
  print(f"Files saved at {out_dir}")
70
  print('finish !')
 
64
  restored = img_as_ubyte(restored[0])
65
 
66
  f = os.path.splitext(os.path.split(file_)[-1])[0]
67
+ save_img((os.path.join(out_dir, '1.png')), restored)
68
 
69
  print(f"Files saved at {out_dir}")
70
  print('finish !')