Create utils.py

utils.py

@@ -0,0 +1,32 @@
+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