Hugging Face's logo

Spaces:
52Hz
/
CMFNet_deraindrop
Running

App Files Community
52Hz commited on
Commit
a2ecb84
·
1 Parent(s): 306a5fa

Update main_test_CMFNet.py

Browse files
Files changed (1) hide show
  1. main_test_CMFNet.py +1 -0
main_test_CMFNet.py CHANGED
@@ -64,6 +64,7 @@ def main():
64
 
65
  f = os.path.splitext(os.path.split(file_)[-1])[0]
66
  save_img((os.path.join(out_dir, f + '.png')), restored_)
 
67
 
68
 
69
  def save_img(filepath, img):
 
64
 
65
  f = os.path.splitext(os.path.split(file_)[-1])[0]
66
  save_img((os.path.join(out_dir, f + '.png')), restored_)
67
+ print('save image to: ', (os.path.join(out_dir, f + '.png'))
68
 
69
 
70
  def save_img(filepath, img):