Spaces:

52Hz
/

SUNet_AWGN_denoising

Running

52Hz commited on Jan 16, 2022

Commit

84f440e

1 Parent(s): bbc278d

Update main_test_SUNet.py

Files changed (1) hide show

main_test_SUNet.py CHANGED Viewed

@@ -101,7 +101,7 @@ def main():
         save_img((os.path.join(out_dir, f + '.png')), restored)
         clean_folder(inp_dir)
-def save_img(filepath, img):
     cv2.imwrite(filepath, cv2.cvtColor(img, cv2.COLOR_RGB2BGR))

         save_img((os.path.join(out_dir, f + '.png')), restored)
         clean_folder(inp_dir)
+def save_img(filepath, img):#
     cv2.imwrite(filepath, cv2.cvtColor(img, cv2.COLOR_RGB2BGR))