Spaces:

52Hz
/

SUNet_AWGN_denoising

Running

52Hz commited on Aug 5

Commit

5434170

•

1 Parent(s): 3e7c45d

Update main_test_SUNet.py

Files changed (1) hide show

main_test_SUNet.py CHANGED Viewed

@@ -102,6 +102,7 @@ def main():
         f = os.path.splitext(os.path.split(file_)[-1])[0]
         save_img((os.path.join(out_dir, f + '.png')), restored)
         clean_folder(inp_dir)
 def save_img(filepath, img):#

         f = os.path.splitext(os.path.split(file_)[-1])[0]
         save_img((os.path.join(out_dir, f + '.png')), restored)
+        print("out_dir: ", out_dir)
         clean_folder(inp_dir)
 def save_img(filepath, img):#