Spaces:

52Hz
/

SRMNet_thesis

Runtime error

52Hz commited on Mar 19, 2022

Commit

6c66462

1 Parent(s): 4c32a21

Update main_test_SRMNet.py

Files changed (1) hide show

main_test_SRMNet.py CHANGED Viewed

@@ -62,7 +62,7 @@ def main():
         restored = img_as_ubyte(restored[0])
         f = os.path.splitext(os.path.split(file_)[-1])[0]
-        save_img((os.path.join(args.result_dir, '1.png')), restored)
     print('finish !')

         restored = img_as_ubyte(restored[0])
         f = os.path.splitext(os.path.split(file_)[-1])[0]
+        save_img((os.path.join(args.result_dir, f + '.png')), restored)
     print('finish !')