utils.py

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