| import cv2 |
| import os |
| import cv2 |
| import shutil |
| import sys |
| from subprocess import call |
|
|
| def run_cmd(command): |
| try: |
| call(command, shell=True) |
| except KeyboardInterrupt: |
| print("Process interrupted") |
| sys.exit(1) |
|
|
| def Restoration(image): |
| os.makedirs("Temp") |
| os.makedirs("Temp/input") |
| print(type(image)) |
| cv2.imwrite("Temp/input/input_img.png", image) |
|
|
| command = ("python run.py --input_folder " |
| + "Temp/input" |
| + " --output_folder " |
| + "Temp" |
| + " --GPU " |
| + "-1" |
| + " --with_scratch") |
| run_cmd(command) |
|
|
| result = cv2.imread("Temp/final_output/input_img.png") |
| shutil.rmtree("Temp") |
| return result |
