Spaces:
Running
Running
Update main_test_SRMNet.py
Browse files- main_test_SRMNet.py +1 -1
main_test_SRMNet.py
CHANGED
@@ -64,7 +64,7 @@ def main():
|
|
64 |
restored = img_as_ubyte(restored[0])
|
65 |
|
66 |
f = os.path.splitext(os.path.split(file_)[-1])[0]
|
67 |
-
save_img((os.path.join(out_dir,
|
68 |
|
69 |
print(f"Files saved at {out_dir}")
|
70 |
print('finish !')
|
|
|
64 |
restored = img_as_ubyte(restored[0])
|
65 |
|
66 |
f = os.path.splitext(os.path.split(file_)[-1])[0]
|
67 |
+
save_img((os.path.join(out_dir, '1.png')), restored)
|
68 |
|
69 |
print(f"Files saved at {out_dir}")
|
70 |
print('finish !')
|