Spaces:

52Hz
/

CMFNet_deraindrop

Sleeping

52Hz commited on Jan 14, 2022

Commit

1cbc520

1 Parent(s): 2fff597

Update main_test_CMFNet.py

Files changed (1) hide show

main_test_CMFNet.py CHANGED Viewed

@@ -63,8 +63,8 @@ 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_)
-        print('save image to: ', (os.path.join(out_dir, f + '.png')))
 def save_img(filepath, img):

         restored_= img_as_ubyte(restored_[0])
         f = os.path.splitext(os.path.split(file_)[-1])[0]
+        save_img((os.path.join(out_dir, 'out.png')), restored_)
 def save_img(filepath, img):