Spaces:

52Hz
/

CMFNet_deraindrop

Sleeping

52Hz commited on Jan 14, 2022

Commit

620cbcf

1 Parent(s): de33178

Update main_test_CMFNet.py

Files changed (1) hide show

main_test_CMFNet.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(out_dir, f + '.png')), restored_)

         restored = img_as_ubyte(restored[0])
         f = os.path.splitext(os.path.split(file_)[-1])[0]
+        save_img((os.path.join(out_dir, f + '.png')), restored)