Update utils/image_io.py

utils/image_io.py

@@ -1,38 +1,38 @@
 import cv2
-import tempfile
+import uuid
+import os
 
 
-def save_png(img):
+OUTPUT_DIR = "outputs"
+os.makedirs(OUTPUT_DIR, exist_ok=True)
 
-    temp = tempfile.NamedTemporaryFile(
-        delete=False,
-        suffix=".png"
+
+def save_bmp(image):
+
+    filename = os.path.join(
+        OUTPUT_DIR,
+        f"{uuid.uuid4()}.bmp"
     )
 
     cv2.imwrite(
-        temp.name,
-        cv2.cvtColor(
-            img,
-            cv2.COLOR_RGB2BGR
-        )
+        filename,
+        image,
+        [cv2.IMWRITE_BMP_COMPRESSION, 0]
     )
 
-    return temp.name
+    return filename
 
 
-def save_bmp(img):
+def save_png(image):
 
-    temp = tempfile.NamedTemporaryFile(
-        delete=False,
-        suffix=".bmp"
+    filename = os.path.join(
+        OUTPUT_DIR,
+        f"{uuid.uuid4()}.png"
     )
 
     cv2.imwrite(
-        temp.name,
-        cv2.cvtColor(
-            img,
-            cv2.COLOR_RGB2BGR
-        )
+        filename,
+        image
     )
 
-    return temp.name
+    return filename