Spaces:
Running
Running
Update main_test_SUNet.py
Browse files- main_test_SUNet.py +1 -0
main_test_SUNet.py
CHANGED
|
@@ -102,6 +102,7 @@ def main():
|
|
| 102 |
|
| 103 |
f = os.path.splitext(os.path.split(file_)[-1])[0]
|
| 104 |
save_img((os.path.join(out_dir, f + '.png')), restored)
|
|
|
|
| 105 |
clean_folder(inp_dir)
|
| 106 |
|
| 107 |
def save_img(filepath, img):#
|
|
|
|
| 102 |
|
| 103 |
f = os.path.splitext(os.path.split(file_)[-1])[0]
|
| 104 |
save_img((os.path.join(out_dir, f + '.png')), restored)
|
| 105 |
+
print("out_dir: ", out_dir)
|
| 106 |
clean_folder(inp_dir)
|
| 107 |
|
| 108 |
def save_img(filepath, img):#
|