Spaces:
Runtime error
Runtime error
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 !')
|